Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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,
|