Update src/diffusers/pipelines/omnigen/pipeline_omnigen.py

Co-authored-by: hlky <[email protected]>
huggingface · Feb 8, 2025 · 39148c3 · 39148c3
1 parent 7888119
commit 39148c3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/diffusers/pipelines/omnigen/pipeline_omnigen.py b/src/diffusers/pipelines/omnigen/pipeline_omnigen.py
@@ -333,7 +333,7 @@ def __call__(
         use_input_image_size_as_output: bool = False,
         num_images_per_prompt: Optional[int] = 1,
         generator: Optional[Union[torch.Generator, List[torch.Generator]]] = None,
-        latents: Optional[torch.FloatTensor] = None,
+        latents: Optional[torch.Tensor] = None,
         output_type: Optional[str] = "pil",
         return_dict: bool = True,
         attention_kwargs: Optional[Dict[str, Any]] = None,