29 lines
808 B
YAML
29 lines
808 B
YAML
services:
|
|
smollora:
|
|
build: .
|
|
image: smollora
|
|
container_name: smollora
|
|
# runtime: nvidia # uncomment if nvidia-container-toolkit is installed
|
|
deploy:
|
|
resources:
|
|
reservations:
|
|
devices:
|
|
- driver: nvidia
|
|
count: all
|
|
capabilities: [gpu]
|
|
volumes:
|
|
- /srv/smollora/data:/data # processed datasets
|
|
- /srv/smollora/output:/output # LoRA adapter output
|
|
- /srv/smollora/hf-cache:/root/.cache/huggingface # model cache
|
|
environment:
|
|
- MODEL=HuggingFaceTB/SmolLM3-3B
|
|
- DATA_DIR=/data
|
|
- OUTPUT_DIR=/output
|
|
- EPOCHS=3
|
|
- BATCH_SIZE=4
|
|
- LR=2e-4
|
|
- LORA_R=16
|
|
- MAX_LENGTH=4096
|
|
- HF_TOKEN=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
|
|
- SKIP_PREP=0
|