tori29umai commited on
Commit
3515ae5
1 Parent(s): 72dca33
Files changed (1) hide show
  1. app.py +2 -1
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,