Spaces:
Runtime error
Runtime error
Update pattern_analyzer.py
Browse files- pattern_analyzer.py +1 -1
pattern_analyzer.py
CHANGED
|
@@ -12,7 +12,7 @@ class PatternAnalyzer:
|
|
| 12 |
"device_map": "auto",
|
| 13 |
"torch_dtype": torch.float32,
|
| 14 |
"low_cpu_mem_usage": True,
|
| 15 |
-
"max_memory": {"cpu": "
|
| 16 |
"offload_folder": "/tmp/offload"
|
| 17 |
}
|
| 18 |
|
|
|
|
| 12 |
"device_map": "auto",
|
| 13 |
"torch_dtype": torch.float32,
|
| 14 |
"low_cpu_mem_usage": True,
|
| 15 |
+
"max_memory": {"cpu": "8GB"},
|
| 16 |
"offload_folder": "/tmp/offload"
|
| 17 |
}
|
| 18 |
|