Spaces:

tmmdev
/

pattern-analysis

Runtime error

tmmdev commited on Feb 15

Commit

77a41a7

verified ·

1 Parent(s): 2eb9374

Update pattern_analyzer.py

Files changed (1) hide show

pattern_analyzer.py CHANGED Viewed

@@ -12,7 +12,7 @@ class PatternAnalyzer:
             "device_map": "auto",
             "torch_dtype": torch.float32,
             "low_cpu_mem_usage": True,
-            "max_memory": {"cpu": "4GB"},
             "offload_folder": "/tmp/offload"
         }

             "device_map": "auto",
             "torch_dtype": torch.float32,
             "low_cpu_mem_usage": True,
+            "max_memory": {"cpu": "8GB"},
             "offload_folder": "/tmp/offload"
         }