multimodalart HF staff commited on
Commit
11e6e67
1 Parent(s): f183407

Update from IF to XL (#2)

Browse files

- Update from IF to XL (a38a8db58376b496722d94c685561792e09bc6f2)

Files changed (1) hide show
  1. model.py +1 -1
model.py CHANGED
@@ -38,7 +38,7 @@ class Model:
38
 
39
  def load_weights(self) -> None:
40
  self.pipe = DiffusionPipeline.from_pretrained(
41
- 'DeepFloyd/IF-I-IF-v1.0',
42
  torch_dtype=torch.float16,
43
  variant='fp16',
44
  use_safetensors=True,
 
38
 
39
  def load_weights(self) -> None:
40
  self.pipe = DiffusionPipeline.from_pretrained(
41
+ 'DeepFloyd/IF-I-XL-v1.0',
42
  torch_dtype=torch.float16,
43
  variant='fp16',
44
  use_safetensors=True,