Lifeinhockey commited on
Commit
1b6e48f
·
verified ·
1 Parent(s): 503e3c8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -4
app.py CHANGED
@@ -501,9 +501,8 @@ def infer(
501
 
502
  generator = torch.Generator(device).manual_seed(seed)
503
 
504
- pipe_LCM_default = get_lora_sd_pipeline(lora_dir='lora_man_animestyle', base_model_name_or_path=model_default, dtype=torch_dtype).to(device)
505
- pipe_LCM = pipe_LCM_default
506
- pipe_LCM.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
507
  pipe_LCM.to(device)
508
 
509
  pipe_LCM.load_lora_weights(adapter_id)
@@ -532,7 +531,7 @@ def infer(
532
  generator = torch.Generator(device).manual_seed(seed)
533
 
534
  pipe_DDIMS = StableDiffusionPipeline.from_pretrained(model_default, torch_dtype=torch_dtype).to(device)
535
- pipe_DDIMS.scheduler = DPMSolverMultistepScheduler.from_config(pipe.scheduler.config)
536
 
537
  prompt_embeds = long_prompt_encoder(prompt, pipe_DDIMS.tokenizer, pipe_DDIMS.text_encoder)
538
  negative_prompt_embeds = long_prompt_encoder(negative_prompt, pipe_DDIMS.tokenizer, pipe_DDIMS.text_encoder)
 
501
 
502
  generator = torch.Generator(device).manual_seed(seed)
503
 
504
+ pipe_LCM = get_lora_sd_pipeline(lora_dir='lora_man_animestyle', base_model_name_or_path=model_default, dtype=torch_dtype).to(device)
505
+ pipe_LCM.scheduler = LCMScheduler.from_config(pipe_LCM.scheduler.config)
 
506
  pipe_LCM.to(device)
507
 
508
  pipe_LCM.load_lora_weights(adapter_id)
 
531
  generator = torch.Generator(device).manual_seed(seed)
532
 
533
  pipe_DDIMS = StableDiffusionPipeline.from_pretrained(model_default, torch_dtype=torch_dtype).to(device)
534
+ pipe_DDIMS.scheduler = DPMSolverMultistepScheduler.from_config(pipe_DDIMS.scheduler.config)
535
 
536
  prompt_embeds = long_prompt_encoder(prompt, pipe_DDIMS.tokenizer, pipe_DDIMS.text_encoder)
537
  negative_prompt_embeds = long_prompt_encoder(negative_prompt, pipe_DDIMS.tokenizer, pipe_DDIMS.text_encoder)