| version: '3.8' | |
| services: | |
| tiny_llama_tuning: | |
| build: . | |
| runtime: nvidia | |
| deploy: | |
| resources: | |
| reservations: | |
| devices: | |
| - driver: nvidia | |
| count: all | |
| capabilities: [gpu] | |
| volumes: | |
| - ./data:/app/data | |
| - ./model_output:/app/model_output | |
| - ./merged_tinyllama_logger:/app/merged_tinyllama_logger | |
| command: python3 training_phi2.py |