Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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.
|
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)
|