Spaces:
Runtime error
Runtime error
update to diffuers pipeline
Browse files
app.py
CHANGED
@@ -35,7 +35,8 @@ MAX_IMAGE_SIZE = int(os.getenv("MAX_IMAGE_SIZE", "768"))
|
|
35 |
USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE") == "1"
|
36 |
DTYPE = torch.float32 # torch.float16 works as well, but pictures seem to be a bit worse
|
37 |
|
38 |
-
pipe = DiffusionPipeline.from_pretrained("SimianLuo/LCM_Dreamshaper_v7"
|
|
|
39 |
pipe.to(torch_device="cuda", torch_dtype=DTYPE)
|
40 |
|
41 |
|
|
|
35 |
USE_TORCH_COMPILE = os.getenv("USE_TORCH_COMPILE") == "1"
|
36 |
DTYPE = torch.float32 # torch.float16 works as well, but pictures seem to be a bit worse
|
37 |
|
38 |
+
pipe = DiffusionPipeline.from_pretrained("SimianLuo/LCM_Dreamshaper_v7")
|
39 |
+
# pipe = DiffusionPipeline.from_pretrained("SimianLuo/LCM_Dreamshaper_v7", custom_pipeline="latent_consistency_txt2img", custom_revision="main")
|
40 |
pipe.to(torch_device="cuda", torch_dtype=DTYPE)
|
41 |
|
42 |
|