Spaces:

DeepFloyd
/

IF

Paused

multimodalart HF staff commited on Apr 28, 2023

Commit

11e6e67

•

1 Parent(s): f183407

Update from IF to XL (#2)

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,