Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -501,9 +501,8 @@ def infer(
|
|
501 |
|
502 |
generator = torch.Generator(device).manual_seed(seed)
|
503 |
|
504 |
-
|
505 |
-
pipe_LCM =
|
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(
|
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)
|