File size: 154 Bytes
a86c1a4
 
 
 
 
 
 
 
5cab769
1
2
3
4
5
6
7
8
9
10
import torch

# Model Configuration
MODEL_NAME = "Kwaipilot/HiPO-8B"

# Generation Configuration
MAX_NEW_TOKENS = 1024
TEMPERATURE = 0.6
DO_SAMPLE = True