mattyamonaca commited on
Commit
ca47a69
1 Parent(s): 64a6d23
Files changed (1) hide show
  1. sd_model.py +3 -7
sd_model.py CHANGED
@@ -52,7 +52,6 @@ def generate(pipe, detectors, prompt, negative_prompt, reference_flg=False, refe
52
 
53
 
54
  if reference_flg==False:
55
- print("####False####")
56
  image = pipe(
57
  prompt=prompt,
58
  negative_prompt = negative_prompt,
@@ -61,13 +60,10 @@ def generate(pipe, detectors, prompt, negative_prompt, reference_flg=False, refe
61
  controlnet_conditioning_scale=[1.0, 0.2],
62
  ).images[0]
63
  else:
64
- print("####True####")
65
- print(reference_img)
66
  pipe.load_ip_adapter(
67
- "h94/IP-Adapter",
68
- subfolder="sdxl_models",
69
- weight_name="ip-adapter-plus_sdxl_vit-h.bin"
70
- )
71
  image = pipe(
72
  prompt=prompt,
73
  negative_prompt = negative_prompt,
 
52
 
53
 
54
  if reference_flg==False:
 
55
  image = pipe(
56
  prompt=prompt,
57
  negative_prompt = negative_prompt,
 
60
  controlnet_conditioning_scale=[1.0, 0.2],
61
  ).images[0]
62
  else:
 
 
63
  pipe.load_ip_adapter(
64
+ "ozzygt/sdxl-ip-adapter",
65
+ "",
66
+ weight_name="ip-adapter_sdxl_vit-h.safetensors")
 
67
  image = pipe(
68
  prompt=prompt,
69
  negative_prompt = negative_prompt,