File size: 310 Bytes
ac9c113 |
1 2 3 4 5 6 7 8 9 10 |
import os
# Configuration for the CyberAttackDetection model
class Config:
MODEL_NAME = "Canstralian/CyberAttackDetection"
TOKENIZER_NAME = "Canstralian/CyberAttackDetection"
MAX_LENGTH = 512
DEVICE = "cuda" if os.environ.get("USE_CUDA", "1") == "1" else "cpu"
PROMPTS_FILE = "prompts.py"
|