Spaces:

tmmdev
/

pattern-analysis

Runtime error

Update pattern_analyzer.py

by tmmdev - opened Feb 17

←

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": {