Spaces:

DeepFloyd
/

IF

Paused

Update from IF to XL

by multimodalart HF staff - opened Apr 28, 2023

←

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,