Lifeinhockey commited on
Commit
5cac376
·
verified ·
1 Parent(s): 3cd2f32

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -106,7 +106,6 @@ pipe_default = get_lora_sd_pipeline(lora_dir='lora_man_animestyle', base_model_n
106
 
107
  # ----------------------------------------------------------------------------------------------------------------------------------------------------
108
  def infer(
109
- pipe_default,
110
  prompt,
111
  negative_prompt,
112
  model=model_default,
@@ -501,6 +500,7 @@ def infer(
501
 
502
  generator = torch.Generator(device).manual_seed(seed)
503
 
 
504
  pipe = pipe_default
505
  pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
506
  pipe.to(device)
@@ -832,7 +832,6 @@ with gr.Blocks(css=css) as demo:
832
  triggers=[run_button.click, prompt.submit],
833
  fn=infer,
834
  inputs=[
835
- pipe_default,
836
  prompt,
837
  negative_prompt,
838
  model,
 
106
 
107
  # ----------------------------------------------------------------------------------------------------------------------------------------------------
108
  def infer(
 
109
  prompt,
110
  negative_prompt,
111
  model=model_default,
 
500
 
501
  generator = torch.Generator(device).manual_seed(seed)
502
 
503
+ pipe_default = get_lora_sd_pipeline(lora_dir='lora_man_animestyle', base_model_name_or_path=model_default, dtype=torch_dtype).to(device)
504
  pipe = pipe_default
505
  pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config)
506
  pipe.to(device)
 
832
  triggers=[run_button.click, prompt.submit],
833
  fn=infer,
834
  inputs=[
 
835
  prompt,
836
  negative_prompt,
837
  model,