Update app_demo.py
Browse files- app_demo.py +2 -2
app_demo.py
CHANGED
|
@@ -32,7 +32,7 @@ DTYPE = torch.float32 # torch.float16 works as well, but pictures seem to be a
|
|
| 32 |
|
| 33 |
#"SimianLuo/LCM_Dreamshaper_v7"
|
| 34 |
'''
|
| 35 |
-
pipe = DiffusionPipeline.from_pretrained( "
|
| 36 |
custom_revision="main",
|
| 37 |
low_cpu_mem_usage=True,
|
| 38 |
safety_checker= None,
|
|
@@ -45,7 +45,7 @@ pipe.to("cpu")
|
|
| 45 |
|
| 46 |
#from diffusers import StableDiffusionPipeline
|
| 47 |
|
| 48 |
-
model_id = "SimianLuo/LCM_Dreamshaper_v7"
|
| 49 |
|
| 50 |
pipe = StableDiffusionPipeline.from_pretrained(model_id)
|
| 51 |
prompt = "A futuristic cityscape at sunset"
|
|
|
|
| 32 |
|
| 33 |
#"SimianLuo/LCM_Dreamshaper_v7"
|
| 34 |
'''
|
| 35 |
+
pipe = DiffusionPipeline.from_pretrained( "openskyml/lcm-lora-sdxl-turbo" , custom_pipeline="openskyml/lcm-lora-sdxl-turbo",
|
| 36 |
custom_revision="main",
|
| 37 |
low_cpu_mem_usage=True,
|
| 38 |
safety_checker= None,
|
|
|
|
| 45 |
|
| 46 |
#from diffusers import StableDiffusionPipeline
|
| 47 |
|
| 48 |
+
model_id = "openskyml/lcm-lora-sdxl-turbo" #"SimianLuo/LCM_Dreamshaper_v7"
|
| 49 |
|
| 50 |
pipe = StableDiffusionPipeline.from_pretrained(model_id)
|
| 51 |
prompt = "A futuristic cityscape at sunset"
|