Spaces:

tmmdev
/

pattern-analysis

Runtime error

tmmdev commited on 7 days ago

Commit

b432788

verified ·

1 Parent(s): f4ebb84

Update pattern_analyzer.py

Files changed (1) hide show

pattern_analyzer.py CHANGED Viewed

@@ -16,7 +16,7 @@ class PatternAnalyzer:
         )
         model_kwargs = {
-            "device_map": "auto",
             "torch_dtype": torch.float32,
             "low_cpu_mem_usage": True,
             "max_memory": {

         )
         model_kwargs = {
+            "device_map": "cpu",
             "torch_dtype": torch.float32,
             "low_cpu_mem_usage": True,
             "max_memory": {