| version: '3.8' | |
| services: | |
| huggingface-android: | |
| build: . | |
| image: huggingface-android:latest | |
| container_name: huggingface-android | |
| environment: | |
| - HUGGINGFACE_TOKEN=${HUGGINGFACE_TOKEN} | |
| volumes: | |
| - ./models:/workspace/models | |
| - ./android-project:/workspace/android-project | |
| - ~/.gradle:/root/.gradle | |
| deploy: | |
| resources: | |
| reservations: | |
| devices: | |
| - driver: nvidia | |
| count: all | |
| capabilities: [gpu] | |
| tty: true | |
| stdin_open: true |