primerz commited on
Commit
0df2aa6
·
verified ·
1 Parent(s): f3238f2

Update generator.py

Browse files
Files changed (1) hide show
  1. generator.py +5 -5
generator.py CHANGED
@@ -81,25 +81,25 @@ class Generator:
81
 
82
  # Create keypoint image
83
  face_kps = draw_kps(processed_image, face_info['kps'])
84
- face_kps_guidance_end = 0.3
85
 
86
  # Set strengths (using 0.8 from file's example)
87
- controlnet_conditioning_scale = [0.8, depth_strength, lineart_strength]
88
- self.mh.pipeline.set_ip_adapter_scale(0.8)
89
  else:
90
  print("No face detected: Disabling InstantID.")
91
  # Create dummy embedding
92
  face_emb = torch.zeros((1, 512), dtype=Config.DTYPE, device=Config.DEVICE)
93
  # Create dummy keypoint image (black)
94
  face_kps = Image.new('RGB', (target_width, target_height), (0, 0, 0))
95
- face_kps_guidance_end = 0.0
96
 
97
  # Set strengths
98
  controlnet_conditioning_scale = [0.0, depth_strength, lineart_strength]
99
  self.mh.pipeline.set_ip_adapter_scale(0.0)
100
 
101
  # We keep the guidance_end for pose low
102
- control_guidance_end = [face_kps_guidance_end, 0.6, 0.6]
103
 
104
  # --- Seed/Generator Logic ---
105
  if seed == -1 or seed is None:
 
81
 
82
  # Create keypoint image
83
  face_kps = draw_kps(processed_image, face_info['kps'])
84
+ face_kps_guidance_end = 0.333
85
 
86
  # Set strengths (using 0.8 from file's example)
87
+ controlnet_conditioning_scale = [0.777, depth_strength, lineart_strength]
88
+ self.mh.pipeline.set_ip_adapter_scale(0.777)
89
  else:
90
  print("No face detected: Disabling InstantID.")
91
  # Create dummy embedding
92
  face_emb = torch.zeros((1, 512), dtype=Config.DTYPE, device=Config.DEVICE)
93
  # Create dummy keypoint image (black)
94
  face_kps = Image.new('RGB', (target_width, target_height), (0, 0, 0))
95
+ face_kps_guidance_end = 0.001
96
 
97
  # Set strengths
98
  controlnet_conditioning_scale = [0.0, depth_strength, lineart_strength]
99
  self.mh.pipeline.set_ip_adapter_scale(0.0)
100
 
101
  # We keep the guidance_end for pose low
102
+ control_guidance_end = [face_kps_guidance_end, 0.666, 0.666]
103
 
104
  # --- Seed/Generator Logic ---
105
  if seed == -1 or seed is None: