Upload LlamaForCausalLM
Browse files- generation_config.json +1 -1
generation_config.json
CHANGED
|
@@ -3,7 +3,7 @@
|
|
| 3 |
"eos_token_id": 2,
|
| 4 |
"max_length": 4096,
|
| 5 |
"max_memory": {
|
| 6 |
-
"cpu":
|
| 7 |
},
|
| 8 |
"no_split_module_classes": [
|
| 9 |
"LlamaDecoderLayer"
|
|
|
|
| 3 |
"eos_token_id": 2,
|
| 4 |
"max_length": 4096,
|
| 5 |
"max_memory": {
|
| 6 |
+
"cpu": 27352305664
|
| 7 |
},
|
| 8 |
"no_split_module_classes": [
|
| 9 |
"LlamaDecoderLayer"
|