Update config.py
Browse files
config.py
CHANGED
|
@@ -1,4 +1,3 @@
|
|
| 1 |
-
```python
|
| 2 |
import torch
|
| 3 |
|
| 4 |
# Model Configuration
|
|
@@ -7,5 +6,4 @@ MODEL_NAME = "Kwaipilot/KAT-Dev"
|
|
| 7 |
# Generation Configuration
|
| 8 |
MAX_NEW_TOKENS = 1024
|
| 9 |
TEMPERATURE = 0.7
|
| 10 |
-
DO_SAMPLE = True
|
| 11 |
-
```
|
|
|
|
|
|
|
| 1 |
import torch
|
| 2 |
|
| 3 |
# Model Configuration
|
|
|
|
| 6 |
# Generation Configuration
|
| 7 |
MAX_NEW_TOKENS = 1024
|
| 8 |
TEMPERATURE = 0.7
|
| 9 |
+
DO_SAMPLE = True
|
|
|