Spaces:
Runtime error
Runtime error
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
|