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"
|