hotfix
Browse files
app.py
CHANGED
|
@@ -9,9 +9,9 @@ from diffusers import StableDiffusionImg2ImgPipeline
|
|
| 9 |
device = "cpu"#"cuda"
|
| 10 |
model_id_or_path = "CompVis/stable-diffusion-v1-4"
|
| 11 |
pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
|
| 12 |
-
model_id_or_path,
|
| 13 |
-
revision = "fp32" #fp16"
|
| 14 |
-
torch_dtype = torch.float32#torch.float16,
|
| 15 |
use_auth_token = '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'
|
| 16 |
)
|
| 17 |
# or download via git clone https://huggingface.co/CompVis/stable-diffusion-v1-4
|
|
|
|
| 9 |
device = "cpu"#"cuda"
|
| 10 |
model_id_or_path = "CompVis/stable-diffusion-v1-4"
|
| 11 |
pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
|
| 12 |
+
model_id_or_path,
|
| 13 |
+
revision = "fp32", #fp16"б
|
| 14 |
+
torch_dtype = torch.float32, #torch.float16,
|
| 15 |
use_auth_token = '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'
|
| 16 |
)
|
| 17 |
# or download via git clone https://huggingface.co/CompVis/stable-diffusion-v1-4
|