Spaces:
Runtime error
Runtime error
downgrade trans
Browse files- Dockerfile +4 -0
- requirements.txt +1 -1
Dockerfile
CHANGED
@@ -17,6 +17,10 @@ ENV PATH="/root/.cargo/bin:${PATH}"
|
|
17 |
RUN pip install --upgrade pip && \
|
18 |
pip install --no-cache-dir -r requirements.txt
|
19 |
|
|
|
|
|
|
|
|
|
20 |
# Create directory and download model
|
21 |
RUN mkdir -p /app/medllama2_7b && \
|
22 |
chmod -R 777 /app/medllama2_7b && \
|
|
|
17 |
RUN pip install --upgrade pip && \
|
18 |
pip install --no-cache-dir -r requirements.txt
|
19 |
|
20 |
+
# Install accelerate and bitsandbytes
|
21 |
+
RUN pip install accelerate
|
22 |
+
RUN pip install -i https://test.pypi.org/simple/ bitsandbytes
|
23 |
+
|
24 |
# Create directory and download model
|
25 |
RUN mkdir -p /app/medllama2_7b && \
|
26 |
chmod -R 777 /app/medllama2_7b && \
|
requirements.txt
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
gradio
|
2 |
-
transformers
|
3 |
accelerate
|
4 |
bitsandbytes
|
5 |
peft
|
|
|
1 |
gradio
|
2 |
+
transformers==4.30
|
3 |
accelerate
|
4 |
bitsandbytes
|
5 |
peft
|