Update generator.py

generator.py

@@ -31,7 +31,8 @@ class Generator:
         input_image, 
         user_prompt="",
         negative_prompt="",
-        guidance_scale=1.5,
+        # --- CHANGED: Default guidance_scale to 0.0 for TCD ---
+        guidance_scale=0.0, 
         num_inference_steps=6,
         img2img_strength=0.3,
         depth_strength=0.3,
@@ -40,6 +41,7 @@ class Generator:
     ):
         # 1. Pre-process Inputs
         print("Processing Input...")
+        # Reverted to original aspect-ratio preserving resize
         processed_image = resize_image_to_1mp(input_image)
         target_width, target_height = processed_image.size
         
@@ -58,6 +60,7 @@ class Generator:
             final_prompt = f"{Config.STYLE_TRIGGER}, {user_prompt}"
             
         print(f"Prompt: {final_prompt}")
+        print(f"Negative Prompt: {negative_prompt}")
 
         # 4. Generate Control Maps
         print("Generating Control Maps (Depth, LineArt)...")
@@ -111,9 +114,9 @@ class Generator:
             control_guidance_end=control_guidance_end,
             clip_skip=2,
             
-            # --- FIX: Set eta to 0.0 to remove stochastic noise ---
-            eta=0.0, 
-            # ------------------------------------------------------
+            # --- TCD Specific Parameter ---
+            eta=0.3, 
+            # ------------------------------
             
         ).images[0]