tmmdev commited on
Commit
ab4ce46
·
verified ·
1 Parent(s): f4ebb84

Update pattern_analyzer.py

Browse files
Files changed (1) hide show
  1. pattern_analyzer.py +2 -2
pattern_analyzer.py CHANGED
@@ -16,7 +16,7 @@ class PatternAnalyzer:
16
  )
17
 
18
  model_kwargs = {
19
- "device_map": "auto",
20
  "torch_dtype": torch.float32,
21
  "low_cpu_mem_usage": True,
22
  "max_memory": {
@@ -91,4 +91,4 @@ class PatternAnalyzer:
91
  })
92
  return patterns
93
  except json.JSONDecodeError:
94
- return []
 
16
  )
17
 
18
  model_kwargs = {
19
+ "device_map": "cpu",
20
  "torch_dtype": torch.float32,
21
  "low_cpu_mem_usage": True,
22
  "max_memory": {
 
91
  })
92
  return patterns
93
  except json.JSONDecodeError:
94
+ return []