hotfix2
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ 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 = "
|
14 |
torch_dtype = torch.float32, #torch.float16,
|
15 |
use_auth_token = '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'
|
16 |
)
|
|
|
10 |
model_id_or_path = "CompVis/stable-diffusion-v1-4"
|
11 |
pipe = StableDiffusionImg2ImgPipeline.from_pretrained(
|
12 |
model_id_or_path,
|
13 |
+
revision = "fp16",
|
14 |
torch_dtype = torch.float32, #torch.float16,
|
15 |
use_auth_token = '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'
|
16 |
)
|