Hugging Face's logo

Spaces:
TheAIBoi
/
ImageGen
Running on Zero

App Files Community
TheAIBoi commited on
Commit
ad5eb1d
·
verified ·
1 Parent(s): 902b92c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ if torch.cuda.is_available():
15
  else:
16
  torch_dtype = torch.float32
17
 
18
- pipe = StableDiffusionXLPipeline.from_pretrained(model_repo_id, vae=vae, torch_dtype=torch_dtype)
19
  pipe = pipe.to(device)
20
 
21
  MAX_SEED = np.iinfo(np.int32).max
 
15
  else:
16
  torch_dtype = torch.float32
17
 
18
+ pipe = StableDiffusionXLPipeline.from_pretrained(model_repo_id, vae=vae, torch_dtype=torch_dtype, use_safetensors=True)
19
  pipe = pipe.to(device)
20
 
21
  MAX_SEED = np.iinfo(np.int32).max