JingyeChen commited on
Commit
933272d
1 Parent(s): 1781379
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -90,13 +90,13 @@ unet = UNet2DConditionModel.from_pretrained(
90
  text_encoder.resize_token_embeddings(len(tokenizer))
91
 
92
 
93
- #### load lcm components
94
- model_id = "lambdalabs/sd-pokemon-diffusers"
95
- lcm_lora_id = "latent-consistency/lcm-lora-sdv1-5"
96
- pipe = DiffusionPipeline.from_pretrained(model_id, unet=unet, tokenizer=tokenizer, text_encoder=text_encoder, torch_dtype=torch.float16)
97
- pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
98
- pipe.load_lora_weights(lcm_lora_id)
99
- pipe.to(device="cuda")
100
 
101
 
102
  #### for interactive
 
90
  text_encoder.resize_token_embeddings(len(tokenizer))
91
 
92
 
93
+ # #### load lcm components
94
+ # model_id = "lambdalabs/sd-pokemon-diffusers"
95
+ # lcm_lora_id = "latent-consistency/lcm-lora-sdv1-5"
96
+ # pipe = DiffusionPipeline.from_pretrained(model_id, unet=unet, tokenizer=tokenizer, text_encoder=text_encoder, torch_dtype=torch.float16)
97
+ # pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
98
+ # pipe.load_lora_weights(lcm_lora_id)
99
+ # pipe.to(device="cuda")
100
 
101
 
102
  #### for interactive