File size: 179 Bytes
d6f9a33
 
 
 
 
1
2
3
4
5
6
import os

MODEL_NAME = "deepseek-ai/deepseek-llm-r1-7b"
HF_TOKEN = os.getenv("HF_TOKEN")  # Set this in the Hugging Face environment
DEVICE = "cuda"  # Change to "cpu" if no GPU