Julian Bilcke
commited on
Commit
·
16386fc
1
Parent(s):
9717c3d
testing something
Browse files- Dockerfile +2 -1
- test.py +2 -0
Dockerfile
CHANGED
|
@@ -58,7 +58,8 @@ RUN pip install -r requirements.txt
|
|
| 58 |
|
| 59 |
COPY --chown=user . .
|
| 60 |
|
| 61 |
-
|
|
|
|
| 62 |
|
| 63 |
# help Pythonia by giving it the path to Python
|
| 64 |
ENV PYTHON_BIN /usr/bin/python3
|
|
|
|
| 58 |
|
| 59 |
COPY --chown=user . .
|
| 60 |
|
| 61 |
+
# temporary skip model download, to make things faster
|
| 62 |
+
# RUN git clone https://huggingface.co/WizardLM/WizardCoder-15B-V1.0
|
| 63 |
|
| 64 |
# help Pythonia by giving it the path to Python
|
| 65 |
ENV PYTHON_BIN /usr/bin/python3
|
test.py
CHANGED
|
@@ -19,6 +19,8 @@ try:
|
|
| 19 |
except:
|
| 20 |
pass
|
| 21 |
|
|
|
|
|
|
|
| 22 |
base_model = "./models/WizardCoder-15B-V1.0"
|
| 23 |
load_8bit = False
|
| 24 |
|
|
|
|
| 19 |
except:
|
| 20 |
pass
|
| 21 |
|
| 22 |
+
print("device: " + device)
|
| 23 |
+
|
| 24 |
base_model = "./models/WizardCoder-15B-V1.0"
|
| 25 |
load_8bit = False
|
| 26 |
|