zero_shot_classification

Sleeping

HFHAB commited on Apr 12

Commit

d2c1b4b

•

1 Parent(s): 6fc1f3d

Update Dockerfile

RUN pip install -U bitsandbytes-cuda110

Files changed (1) hide show

Dockerfile CHANGED Viewed

@@ -5,7 +5,7 @@ WORKDIR /code
 COPY ./requirements.txt /code/requirements.txt
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
-RUN pip install -U bitsandbytes
 COPY . .

 COPY ./requirements.txt /code/requirements.txt
 RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+RUN pip install -U bitsandbytes-cuda110
 COPY . .