Fix load local config for non-A100 setup (#100)

pytorch-labs · Dec 8, 2023 · d40f920 · d40f920
1 parent ceaae4b
commit d40f920
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/segment_anything_fast/flash_4.py b/segment_anything_fast/flash_4.py
@@ -234,7 +234,6 @@ def _load_best_configs():
         saved_configs = cwd / "flash_4_configs.p"
         print(f"We will try to read previously created kernel configurations from {saved_configs}.")
         print("You can disable this kernel by setting SEGMENT_ANYTHING_FAST_USE_FLASH_4=0")
-        return None
     if saved_configs.is_file():
         import pickle
         with open(saved_configs, 'rb') as f: