Create Dockerfile
Browse files- Dockerfile +7 -0
Dockerfile
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
FROM ghcr.io/huggingface/text-generation-inference:latest
|
| 2 |
+
|
| 3 |
+
# Install dependencies
|
| 4 |
+
RUN pip install --no-cache-dir torch torchvision torchaudio transformers accelerate bitsandbytes
|
| 5 |
+
|
| 6 |
+
# Pull model dari Hugging Face
|
| 7 |
+
RUN text-generation-launcher --model TheBloke/MythoMax-L2-13B-GGUF
|