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
|