primerz commited on
Commit
ab4a566
·
verified ·
1 Parent(s): b96a00f

Update generator.py

Browse files
Files changed (1) hide show
  1. generator.py +3 -6
generator.py CHANGED
@@ -40,7 +40,6 @@ class Generator:
40
  ):
41
  # 1. Pre-process Inputs
42
  print("Processing Input...")
43
- # Reverted to original aspect-ratio preserving resize
44
  processed_image = resize_image_to_1mp(input_image)
45
  target_width, target_height = processed_image.size
46
 
@@ -104,19 +103,17 @@ class Generator:
104
  image_embeds=face_emb,
105
  generator=generator,
106
 
107
- # --- Static Values (Adaptive Logic Removed) ---
108
  strength=img2img_strength,
109
  guidance_scale=guidance_scale,
110
  num_inference_steps=num_inference_steps,
111
- # --------------------------------------------
112
 
113
  controlnet_conditioning_scale=controlnet_conditioning_scale,
114
  control_guidance_end=control_guidance_end,
115
  clip_skip=2,
116
 
117
- # --- TCD Specific Parameter ---
118
- eta=0.3,
119
- # ------------------------------
120
 
121
  ).images[0]
122
 
 
40
  ):
41
  # 1. Pre-process Inputs
42
  print("Processing Input...")
 
43
  processed_image = resize_image_to_1mp(input_image)
44
  target_width, target_height = processed_image.size
45
 
 
103
  image_embeds=face_emb,
104
  generator=generator,
105
 
 
106
  strength=img2img_strength,
107
  guidance_scale=guidance_scale,
108
  num_inference_steps=num_inference_steps,
 
109
 
110
  controlnet_conditioning_scale=controlnet_conditioning_scale,
111
  control_guidance_end=control_guidance_end,
112
  clip_skip=2,
113
 
114
+ # --- FIX: Set eta to 0.0 to remove stochastic noise ---
115
+ eta=0.0,
116
+ # ------------------------------------------------------
117
 
118
  ).images[0]
119