gokaygokay commited on
Commit
2e90707
·
1 Parent(s): 8c8a286

fix models

Browse files
Files changed (1) hide show
  1. llm_inference_video.py +9 -1
llm_inference_video.py CHANGED
@@ -53,6 +53,8 @@ class VideoLLMInferenceNode:
53
  camera_movement = camera_direction
54
 
55
  # Video prompt templates
 
 
56
  prompt_templates = {
57
  "minimalist": f"""Create an elegantly sparse video description focusing on {input_concept}.
58
  {format_element(camera_movement, 'camera')}
@@ -79,7 +81,13 @@ class VideoLLMInferenceNode:
79
  "experimental": f"""Create an avant-garde paragraph describing an experimental video exploring {input_concept}. Embrace unconventional storytelling through {style} aesthetics, {camera_style} techniques, and {pacing} temporal flow. Incorporate {special_effects} digital manipulations and {custom_elements if custom_elements else 'abstract visual metaphors'} to challenge traditional narrative structures."""
80
  }
81
 
82
- base_prompt = prompt_templates.get(style.lower(), prompt_templates["cinematic"])
 
 
 
 
 
 
83
 
84
  # Configure length requirements
85
  length_config = {
 
53
  camera_movement = camera_direction
54
 
55
  # Video prompt templates
56
+ default_style = "simple" # Changed from "cinematic" to "simple" as default
57
+
58
  prompt_templates = {
59
  "minimalist": f"""Create an elegantly sparse video description focusing on {input_concept}.
60
  {format_element(camera_movement, 'camera')}
 
81
  "experimental": f"""Create an avant-garde paragraph describing an experimental video exploring {input_concept}. Embrace unconventional storytelling through {style} aesthetics, {camera_style} techniques, and {pacing} temporal flow. Incorporate {special_effects} digital manipulations and {custom_elements if custom_elements else 'abstract visual metaphors'} to challenge traditional narrative structures."""
82
  }
83
 
84
+ # Get the template with a more neutral default
85
+ selected_style = style.lower()
86
+ if selected_style not in prompt_templates:
87
+ print(f"Warning: Style '{style}' not found, using '{default_style}' template")
88
+ selected_style = default_style
89
+
90
+ base_prompt = prompt_templates[selected_style]
91
 
92
  # Configure length requirements
93
  length_config = {