SAM2 AMG load_fast fix (#1374)

pytorch · Dec 3, 2024 · b7630f1 · b7630f1
1 parent ebb9086
commit b7630f1
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/examples/sam2_amg_server/server.py b/examples/sam2_amg_server/server.py
@@ -588,6 +588,12 @@ def main(checkpoint_path,
     if load_fast != "":
         load_aot_fast(mask_generator, load_fast)
 
+    if furious:
+        set_furious(mask_generator)
+    # since autoquant is replicating what furious mode is doing, don't use these two together
+    elif use_autoquant:
+        set_autoquant(mask_generator)
+
     if save_fast != "":
         assert load_fast == "", "Can't save compiled models while loading them with --load-fast."
         assert not baseline, "--fast cannot be combined with baseline. code to be torch.compile(fullgraph=True) compatible."
@@ -598,12 +604,6 @@ def main(checkpoint_path,
         assert not baseline, "--fast cannot be combined with baseline. code to be torch.compile(fullgraph=True) compatible."
         set_fast(mask_generator, load_fast)
 
-    if furious:
-        set_furious(mask_generator)
-    # since autoquant is replicating what furious mode is doing, don't use these two together
-    elif use_autoquant:
-        set_autoquant(mask_generator)
-
     with open('dog.jpg', 'rb') as f:
         image_tensor = file_bytes_to_image_tensor(bytearray(f.read()))