charliebaby2023 commited on
Commit
8bf9568
·
verified ·
1 Parent(s): fd1718e

Update app_demo.py

Browse files
Files changed (1) hide show
  1. 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( "SimianLuo/LCM_Dreamshaper_v7" , custom_pipeline="SimianLuo/LCM_Dreamshaper_v7",
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"