Antonio Toro Jaén commited on
Commit
f161c43
·
1 Parent(s): 6dea781
Files changed (2) hide show
  1. Dockerfile +17 -0
  2. requirements.txt +0 -7
Dockerfile ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ FROM huggingface/transformers-pytorch-gpu:latest
2
+
3
+ # Instala dependencias del sistema
4
+ RUN apt-get update && apt-get install -y git
5
+
6
+ # Actualiza pip
7
+ RUN pip install --upgrade pip
8
+
9
+ # Instala tus dependencias de Python
10
+ RUN pip install \
11
+ gradio \
12
+ bitsandbytes \
13
+ transformers \
14
+ accelerate \
15
+ peft \
16
+ einops \
17
+ scipy
requirements.txt DELETED
@@ -1,7 +0,0 @@
1
- gradio
2
- torch
3
- transformers
4
- unsloth
5
- accelerate
6
- bitsandbytes
7
- scipy # requerido por algunos backends de HF