Hugging Face's logo

Spaces:
Rahul8827
/
text2image22
Runtime error

App Files Community
Rahul8827 commited on
Commit
fb1a904
•
1 Parent(s): b329693

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ class CFG:
13
  image_gen_guidance_scale = 9
14
 
15
  image_gen_model = StableDiffusionPipeline.from_pretrained(
16
- CFG.image_gen_model_id, torch_dtype=torch.float16,
17
  revision="fp16", use_auth_token='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', guidance_scale=9
18
  )
19
  apply = image_gen_model.to(CFG.device)
 
13
  image_gen_guidance_scale = 9
14
 
15
  image_gen_model = StableDiffusionPipeline.from_pretrained(
16
+ CFG.image_gen_model_id, torch_dtype=torch.float32,
17
  revision="fp16", use_auth_token='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', guidance_scale=9
18
  )
19
  apply = image_gen_model.to(CFG.device)