Update test_stablehairv2.py

test_stablehairv2.py

@@ -100,10 +100,11 @@ def log_validation(
     gscale = float(_os.getenv('SH_GUIDANCE', getattr(args, 'guidance_scale', 1.5)))
     vlen = int(_os.getenv('SH_VIDEO_LENGTH', getattr(args, 'video_length', 21)))
     # 统一时序长度：上下文帧数始终等于视频帧数（不再读取 SH_CONTEXT_FRAMES）
-    cframes = vlen
-    print("推理步数：",steps)
-    print("guidance_scale:",gscale)
-    print("视频帧数：",vlen)
+    cframes = int(_os.getenv('SH_CFRAMES', getattr(args, 'cframes', 12)))
+    print("[cfg]推理步数：",steps)
+    print("[cfg]guidance_scale:",gscale)
+    print("[cfg]视频帧数：",vlen)
+    print("[cfg]cframes:",cframes)
     # Generate camera trajectory with exactly vlen frames
     angles = np.linspace(0, 2 * np.pi, vlen, endpoint=False)
     X = 0.4 * np.sin(angles)
@@ -132,7 +133,7 @@ def log_validation(
     id_image = cv2.cvtColor(cv2.imread(temp_bald_path), cv2.COLOR_BGR2RGB)
     id_image = cv2.resize(id_image, (512, 512))
 
-    id_list = [id_image for _ in range(12)]
+    id_list = [id_image for _ in range(cframes)]
     if align_enabled:
         hair_image = _maybe_align_image(args.validation_hairs[0], output_size=align_size, prefer_cuda=prefer_cuda)
         prompt_img = _maybe_align_image(args.validation_ids[0], output_size=align_size, prefer_cuda=prefer_cuda)
@@ -152,8 +153,8 @@ def log_validation(
         result = pipeline(
             prompt="",
             negative_prompt="",
-            num_inference_steps=30,
-            guidance_scale=1.5,
+            num_inference_steps=steps,
+            guidance_scale=gscale,
             width=512,
             height=512,
             controlnet_condition=id_list,
@@ -165,8 +166,8 @@ def log_validation(
             poses=None,
             x=x_tensor,
             y=y_tensor,
-            video_length=21,
-            context_frames=12,
+            video_length=vlen,
+            context_frames=cframes,
         )
         video = torch.cat([result.videos, result.videos], dim=0)
         video_path = os.path.join(output_dir, f"generated_video_{idx}.mp4")