Spaces:
Sleeping
Sleeping
t5 commited on
Commit ·
ed8e91b
1
Parent(s): 179273d
apple
Browse files- Dockerfile +1 -1
- wr.py +1 -1
Dockerfile
CHANGED
|
@@ -17,7 +17,7 @@ ENV tmp=$HOME/tmp \
|
|
| 17 |
# COPY ./requirements.txt /code/requirements.txt
|
| 18 |
|
| 19 |
RUN python -m pip install --upgrade pip setuptools
|
| 20 |
-
|
| 21 |
RUN pip install --no-cache-dir --upgrade -r requirements.txt
|
| 22 |
RUN pip install gunicorn
|
| 23 |
|
|
|
|
| 17 |
# COPY ./requirements.txt /code/requirements.txt
|
| 18 |
|
| 19 |
RUN python -m pip install --upgrade pip setuptools
|
| 20 |
+
RUN pip install torch==2.0.1 torchvision==0.15.2 --extra-index-url https://download.pytorch.org/whl/cu118
|
| 21 |
RUN pip install --no-cache-dir --upgrade -r requirements.txt
|
| 22 |
RUN pip install gunicorn
|
| 23 |
|
wr.py
CHANGED
|
@@ -1652,7 +1652,7 @@ def xfil(ap,flinput=1):
|
|
| 1652 |
import importlib; from importlib import import_module
|
| 1653 |
def impt(fn,reload=1,**_):
|
| 1654 |
dir,fir=lfn.split(fn)
|
| 1655 |
-
|
| 1656 |
cf=cfile(fn)
|
| 1657 |
if cf in sys.modules:
|
| 1658 |
mod=sys.modules[cf]
|
|
|
|
| 1652 |
import importlib; from importlib import import_module
|
| 1653 |
def impt(fn,reload=1,**_):
|
| 1654 |
dir,fir=lfn.split(fn)
|
| 1655 |
+
sys.path.append(os.path.abspath(dir))
|
| 1656 |
cf=cfile(fn)
|
| 1657 |
if cf in sys.modules:
|
| 1658 |
mod=sys.modules[cf]
|