Fixed safety Inputs to videoInference

2026-06-01 11:32:53 +05:30
parent 03bb544f7d
commit a97eb7c97f
2 changed files with 2 additions and 8 deletions
@@ -422,14 +422,8 @@ class txt2vid:
                genConfig[0]["providerSettings"] = providerSettings
        
        # Add safety inputs if provided
-        # Note: Video inference API only supports 'mode' key for safety
        if safetyInputs is not None and isinstance(safetyInputs, dict) and len(safetyInputs) > 0:
-            # Filter to only include 'mode' (only allowed key for video inference)
-            filtered_safety = {}
-            if "mode" in safetyInputs:
-                filtered_safety["mode"] = safetyInputs["mode"]
-            if filtered_safety:
-                genConfig[0]["safety"] = filtered_safety
+            genConfig[0]["safety"] = safetyInputs
        
        # Add video advanced feature inputs if provided
        if videoAdvancedFeatureInputs is not None and isinstance(videoAdvancedFeatureInputs, dict) and len(videoAdvancedFeatureInputs) > 0: