charliebaby2023 commited on
Commit
aae0a97
·
verified ·
1 Parent(s): 3986b92

Update easy_run.py

Browse files
Files changed (1) hide show
  1. easy_run.py +2 -2
easy_run.py CHANGED
@@ -5,9 +5,9 @@ from lcm_scheduler import LCMScheduler
5
  import hf_image_uploader as hiu
6
  import torch
7
 
8
- scheduler = LCMScheduler.from_pretrained("SimianLuo/LCM_Dreamshaper_v7", subfolder="scheduler")
9
 
10
- pipe = LatentConsistencyModelPipeline.from_pretrained("SimianLuo/LCM_Dreamshaper_v7", scheduler=scheduler)
11
  pipe.to("cuda", dtype=torch.float16)
12
 
13
  prompt = "a red horse"
 
5
  import hf_image_uploader as hiu
6
  import torch
7
 
8
+ scheduler = LCMScheduler.from_pretrained("charliebaby2023/p_y", subfolder="scheduler")
9
 
10
+ pipe = LatentConsistencyModelPipeline.from_pretrained("charliebaby2023/p_y", scheduler=scheduler)
11
  pipe.to("cuda", dtype=torch.float16)
12
 
13
  prompt = "a red horse"