euiia commited on
Commit
eafff49
·
verified ·
1 Parent(s): dda79db

Update engineers/deformes4D.py

Browse files
Files changed (1) hide show
  1. engineers/deformes4D.py +3 -3
engineers/deformes4D.py CHANGED
@@ -28,7 +28,7 @@ from aduc_types import LatentConditioningItem
28
  from managers.ltx_manager import ltx_manager_singleton
29
  from managers.latent_enhancer_manager import latent_enhancer_specialist_singleton
30
  from managers.vae_manager import vae_manager_singleton
31
- from managers.gemini_manager import gemini_manager_singleton
32
  from managers.seedvr_manager import seedvr_manager_singleton
33
  from managers.mmaudio_manager import mmaudio_manager_singleton
34
  from tools.video_encode_tool import video_encode_tool_singleton
@@ -126,8 +126,8 @@ class Deformes4DEngine:
126
  start_keyframe_path = keyframe_paths[i]
127
  destination_keyframe_path = keyframe_paths[i + 1]
128
  future_story_prompt = storyboard[i + 1] if (i + 1) < len(storyboard) else "The final scene."
129
- logger.info(f"Calling Gemini to generate cinematic decision for fragment {fragment_index}...")
130
- decision = gemini_manager_singleton.get_cinematic_decision(global_prompt, story_history, past_keyframe_path, start_keyframe_path, destination_keyframe_path, storyboard[i - 1] if i > 0 else "The beginning.", storyboard[i], future_story_prompt)
131
  transition_type, motion_prompt = decision["transition_type"], decision["motion_prompt"]
132
  story_history += f"\n- Act {fragment_index}: {motion_prompt}"
133
 
 
28
  from managers.ltx_manager import ltx_manager_singleton
29
  from managers.latent_enhancer_manager import latent_enhancer_specialist_singleton
30
  from managers.vae_manager import vae_manager_singleton
31
+ from managers.deformes2D_thinker import deformes2d_thinker_singleton
32
  from managers.seedvr_manager import seedvr_manager_singleton
33
  from managers.mmaudio_manager import mmaudio_manager_singleton
34
  from tools.video_encode_tool import video_encode_tool_singleton
 
126
  start_keyframe_path = keyframe_paths[i]
127
  destination_keyframe_path = keyframe_paths[i + 1]
128
  future_story_prompt = storyboard[i + 1] if (i + 1) < len(storyboard) else "The final scene."
129
+ logger.info(f"Calling deformes2D_thinker to generate cinematic decision for fragment {fragment_index}...")
130
+ decision = deformes2d_thinker_singleton.get_cinematic_decision(global_prompt, story_history, past_keyframe_path, start_keyframe_path, destination_keyframe_path, storyboard[i - 1] if i > 0 else "The beginning.", storyboard[i], future_story_prompt)
131
  transition_type, motion_prompt = decision["transition_type"], decision["motion_prompt"]
132
  story_history += f"\n- Act {fragment_index}: {motion_prompt}"
133