Hugging Face's logo

Spaces:
salehalmansour
/
T2I-Final
No application file

App Files Community
salehalmansour commited on
Commit
666025a
·
verified ·
1 Parent(s): b8f0a25

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,3 +1,4 @@
 
1
  class CFG:
2
  device = "cpu"
3
  seed = 42
@@ -9,7 +10,7 @@ class CFG:
9
 
10
  image_gen_model = StableDiffusionPipeline.from_pretrained(
11
  CFG.image_gen_model_id, torch_dtype=torch.float32,
12
- revision="fp16", guidance_scale=9
13
  )
14
  apply = image_gen_model.to(CFG.device)
15
 
 
1
+
2
  class CFG:
3
  device = "cpu"
4
  seed = 42
 
10
 
11
  image_gen_model = StableDiffusionPipeline.from_pretrained(
12
  CFG.image_gen_model_id, torch_dtype=torch.float32,
13
+ revision="fp16" , guidance_scale=9
14
  )
15
  apply = image_gen_model.to(CFG.device)
16