Commit
·
a4d2baa
1
Parent(s):
66de988
build fix
Browse files- Dockerfile +2 -0
- services/model_loader.py +1 -1
Dockerfile
CHANGED
@@ -3,6 +3,8 @@
|
|
3 |
|
4 |
FROM python:3.12-slim
|
5 |
|
|
|
|
|
6 |
WORKDIR /code
|
7 |
|
8 |
COPY ./requirements.txt /code/requirements.txt
|
|
|
3 |
|
4 |
FROM python:3.12-slim
|
5 |
|
6 |
+
ENV TORCH_HOME=/tmp/torch_cache
|
7 |
+
|
8 |
WORKDIR /code
|
9 |
|
10 |
COPY ./requirements.txt /code/requirements.txt
|
services/model_loader.py
CHANGED
@@ -15,5 +15,5 @@ def load_model(weights_path):
|
|
15 |
pos_enc="absolute"
|
16 |
).to(DEVICE)
|
17 |
# model.load_state_dict(torch.load(weights_path, map_location=DEVICE))
|
18 |
-
model.eval
|
19 |
return model
|
|
|
15 |
pos_enc="absolute"
|
16 |
).to(DEVICE)
|
17 |
# model.load_state_dict(torch.load(weights_path, map_location=DEVICE))
|
18 |
+
model.eval()
|
19 |
return model
|