Spaces:
Running
on
Zero
Running
on
Zero
Update generator.py
Browse files- 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
|