Spaces:
Running
on
Zero
Running
on
Zero
tori29umai
commited on
Commit
•
3515ae5
1
Parent(s):
72dca33
app.py
Browse files
app.py
CHANGED
@@ -38,7 +38,7 @@ def load_model(lora_dir, cn_dir):
|
|
38 |
use_safetensors=True,
|
39 |
scheduler=scheduler,
|
40 |
)
|
41 |
-
pipe.load_lora_weights(lora_dir, weight_name="sdxl_BWLine.safetensors")
|
42 |
pipe = pipe.to(device)
|
43 |
return pipe
|
44 |
|
@@ -59,6 +59,7 @@ def predict(input_image_path, prompt, negative_prompt, controlnet_scale):
|
|
59 |
prompt = execute_prompt(execute_tags, prompt)
|
60 |
prompt = remove_duplicates(prompt)
|
61 |
prompt = remove_color(prompt)
|
|
|
62 |
|
63 |
output_image = pipe(
|
64 |
image=white_base_pil,
|
|
|
38 |
use_safetensors=True,
|
39 |
scheduler=scheduler,
|
40 |
)
|
41 |
+
# pipe.load_lora_weights(lora_dir, weight_name="sdxl_BWLine.safetensors")
|
42 |
pipe = pipe.to(device)
|
43 |
return pipe
|
44 |
|
|
|
59 |
prompt = execute_prompt(execute_tags, prompt)
|
60 |
prompt = remove_duplicates(prompt)
|
61 |
prompt = remove_color(prompt)
|
62 |
+
print(prompt)
|
63 |
|
64 |
output_image = pipe(
|
65 |
image=white_base_pil,
|