primerz commited on
Commit
228348f
·
verified ·
1 Parent(s): 545e006

Update generator.py

Browse files
Files changed (1) hide show
  1. generator.py +5 -4
generator.py CHANGED
@@ -116,10 +116,7 @@ class Generator:
116
  print(f"Face Coverage: {coverage_ratio:.3f} ({int(coverage_ratio * 12)}/12)")
117
 
118
  # 2. Define Control Points (Half Less Aggressive)
119
- # CFG: 0.825 start (17.5% reduction)
120
  cfg_mult = self.solve_bezier(coverage_ratio, 0.825, 0.85, 0.95, 1.0)
121
-
122
- # Strength: 0.9375 start (6.25% reduction)
123
  str_mult = self.solve_bezier(coverage_ratio, 0.9375, 0.95, 0.99, 1.0)
124
 
125
  # 3. Apply Multipliers
@@ -190,13 +187,17 @@ class Generator:
190
  # --- Using Adaptive Values ---
191
  strength=adaptive_strength,
192
  guidance_scale=adaptive_cfg,
193
- num_inference_steps=num_inference_steps,
194
  # ---------------------------
195
 
196
  controlnet_conditioning_scale=controlnet_conditioning_scale,
197
  control_guidance_end=control_guidance_end,
198
  clip_skip=2,
199
 
 
 
 
 
200
  ).images[0]
201
 
202
  return result
 
116
  print(f"Face Coverage: {coverage_ratio:.3f} ({int(coverage_ratio * 12)}/12)")
117
 
118
  # 2. Define Control Points (Half Less Aggressive)
 
119
  cfg_mult = self.solve_bezier(coverage_ratio, 0.825, 0.85, 0.95, 1.0)
 
 
120
  str_mult = self.solve_bezier(coverage_ratio, 0.9375, 0.95, 0.99, 1.0)
121
 
122
  # 3. Apply Multipliers
 
187
  # --- Using Adaptive Values ---
188
  strength=adaptive_strength,
189
  guidance_scale=adaptive_cfg,
190
+ num_inference_steps=num_inference_steps,
191
  # ---------------------------
192
 
193
  controlnet_conditioning_scale=controlnet_conditioning_scale,
194
  control_guidance_end=control_guidance_end,
195
  clip_skip=2,
196
 
197
+ # --- TCD Specific Parameter ---
198
+ eta=0.3, # Controls stochasticity (gamma) for TCD
199
+ # ------------------------------
200
+
201
  ).images[0]
202
 
203
  return result