Spaces:

DeepFloyd
/

IF

Paused

multimodalart HF Staff commited on Apr 28, 2023

Commit

a38a8db

1 Parent(s): f183407

Update from IF to XL

Files changed (1) hide show

model.py CHANGED Viewed

@@ -38,7 +38,7 @@ class Model:
     def load_weights(self) -> None:
         self.pipe = DiffusionPipeline.from_pretrained(
-            'DeepFloyd/IF-I-IF-v1.0',
             torch_dtype=torch.float16,
             variant='fp16',
             use_safetensors=True,

     def load_weights(self) -> None:
         self.pipe = DiffusionPipeline.from_pretrained(
+            'DeepFloyd/IF-I-XL-v1.0',
             torch_dtype=torch.float16,
             variant='fp16',
             use_safetensors=True,