SmolFactory / requirements /requirements_core.txt
Tonic's picture
improves spaces deployment , configuration for custom settings , adds interface for spaces deployment
5f8b28d
# Core dependencies for SmolLM3 and GPT-OSS fine-tuning
torch>=2.0.0
transformers @ git+https://github.com/huggingface/transformers.git # Latest version with GPT-OSS support
datasets>=2.14.0
accelerate>=0.20.0
peft>=0.17.0 # Updated for GPT-OSS LoRA support
trl>=0.20.0 # Updated for GPT-OSS compatibility
kernels
openai-harmony
# Hugging Face Hub for model and space management
huggingface_hub>=0.19.0
# Monitoring and tracking
requests>=2.31.0
numpy>=1.24.0
pandas>=2.0.0
plotly>=5.0.0
trackio>=0.1.0
psutil>=5.9.0
pynvml>=12.0.0
# GPT-OSS specific dependencies
# Note: GPT-OSS requires specific versions for optimal performance
# These are compatible with the tutorial requirements
bitsandbytes>=0.41.0 # For 4-bit quantization
triton >= 3.4.0
kernels