Hugging Face's logo

Spaces:
ehristoforu
/
Mobius
Runtime error

App Files Community
ehristoforu commited on
Commit
ab72545
·
verified ·
1 Parent(s): 641b44e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -41,7 +41,8 @@ vae = AutoencoderKL.from_pretrained(
41
  pipe = StableDiffusionXLPipeline.from_pretrained(
42
  "Corcelio/mobius",
43
  vae=vae,
44
- torch_dtype=torch.float16
 
45
  )
46
  pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)
47
  pipe.to('cuda')
 
41
  pipe = StableDiffusionXLPipeline.from_pretrained(
42
  "Corcelio/mobius",
43
  vae=vae,
44
+ torch_dtype=torch.float16,
45
+ use_safetensors=False,
46
  )
47
  pipe.scheduler = KDPM2AncestralDiscreteScheduler.from_config(pipe.scheduler.config)
48
  pipe.to('cuda')