cuio
/

MiniT2I_bf16

cuio commited on Jun 26

Commit

900ef79

verified ·

1 Parent(s): f33cb1d

Update pipeline.py

Files changed (1) hide show

pipeline.py CHANGED Viewed

@@ -678,7 +678,7 @@ class MiniT2IPipeline(DiffusionPipeline):
         self,
         prompt: Union[str, List[str]],
         model_type: str = "b16",
-        repo_id_or_path: Union[str, os.PathLike] = "dummy9996/MiniT2I_bf16",
         torch_dtype: Optional[torch.dtype] = torch.bfloat16,
         text_encoder_dtype: torch.dtype = torch.float32,
         device: Optional[Union[str, torch.device]] = None,

         self,
         prompt: Union[str, List[str]],
         model_type: str = "b16",
+        repo_id_or_path: Union[str, os.PathLike] = "cuio/MiniT2I_bf16",
         torch_dtype: Optional[torch.dtype] = torch.bfloat16,
         text_encoder_dtype: torch.dtype = torch.float32,
         device: Optional[Union[str, torch.device]] = None,