Spaces:
Running on Zero
Running on Zero
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,
|