unlogged users

app.py

@@ -252,6 +252,8 @@ def is_portrait(video_file):
 
 def calculate_time_required(max_duration_s, rc_bool):
 
+    if max_duration_s == -2:
+        return None
     if max_duration_s == -1:
         return 75
     if max_duration_s == 2:
@@ -307,7 +309,12 @@ def _animate(input_video, max_duration_s, edited_frame, rc_bool, pts_by_frame, l
             w, h = 360, 640
         else:
             w, h = 640, 360
-        
+    elif max_duration_s == -2:
+        if is_portrait(input_video):
+            w, h = 240, 426
+        else:
+            w, h = 426, 240
+
     tag_string = "replace_flag" if rc_bool else "retarget_flag"
 
     preprocess_model = load_preprocess_models(max_duration_s)
@@ -515,17 +522,32 @@ def animate_scene(input_video, max_duration_s, edited_frame, rc_str,
     except Exception as e:
         err = str(e).lower()
         print(f"{session_id} failed due to {err}")
+
+        if "zerogpu quotas" not in err and "pro gpu quota" not in err:
+            raise
+
+        max_duration_s = -1
+
         try:
             pts_by_frame, lbs_by_frame = {}, {}
             
             output_video_path = _animate(
-                input_video, -1, edited_frame_png, rc_bool,
+                input_video, max_duration_s, edited_frame_png, rc_bool,
                 pts_by_frame, lbs_by_frame, session_id, progress
             )
         except Exception as e:
             err = str(e).lower()
             print(f"{session_id} failed due to {err}")
-            raise   
+
+            if "unlogged" in err:
+                max_duration_s = -2
+
+                output_video_path = _animate(
+                    input_video, max_duration_s, edited_frame_png, rc_bool,
+                    pts_by_frame, lbs_by_frame, session_id, progress
+                )
+            else:
+                raise   
 
     final_video_path = os.path.join(output_dir, 'final_result.mp4')
 

wan/modules/animate/preprocess/preprocess_data.py

@@ -39,7 +39,7 @@ def load_preprocess_models(max_duration_s):
     pose2d_checkpoint_path = os.path.join(ckpt_path, 'pose2d/vitpose_h_wholebody.onnx')
     det_checkpoint_path = os.path.join(ckpt_path, 'det/yolov10m.onnx')
 
-    if max_duration_s == -1:
+    if max_duration_s < 0:
         print("using small sam2")
         sam2_checkpoint_path = [os.path.join(ckpt_path, 'sam2/sam2_hiera_small.pt'),"sam2_hiera_s.yaml"]
     else: