AnkitS1997 commited on
Commit
47db460
·
1 Parent(s): 2b0a59c

updated transformer version

Browse files
.ipynb_checkpoints/Dockerfile-checkpoint CHANGED
@@ -12,7 +12,7 @@ RUN pip install --upgrade pip
12
 
13
  RUN pip install --no-cache-dir -r requirements.txt
14
 
15
- RUN pip install --no-cache-dir torch transformers
16
 
17
  # Install supervisor to manage multiple processes
18
  RUN apt-get update && apt-get install -y supervisor \
 
12
 
13
  RUN pip install --no-cache-dir -r requirements.txt
14
 
15
+ RUN pip install --no-cache-dir torch
16
 
17
  # Install supervisor to manage multiple processes
18
  RUN apt-get update && apt-get install -y supervisor \
.ipynb_checkpoints/requirements-checkpoint.txt ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ fastapi==0.111.0
2
+ pillow==10.3.0
3
+ pydantic==1.10.16
4
+ streamlit==1.37.1
5
+ supervisor
6
+ uvicorn==0.30.1
7
+ typing_extensions
8
+ accelerate
9
+ huggingface_hub
10
+ bitsandbytes
11
+ peft
12
+ transformers==4.37.2
Dockerfile CHANGED
@@ -12,7 +12,7 @@ RUN pip install --upgrade pip
12
 
13
  RUN pip install --no-cache-dir -r requirements.txt
14
 
15
- RUN pip install --no-cache-dir torch transformers
16
 
17
  # Install supervisor to manage multiple processes
18
  RUN apt-get update && apt-get install -y supervisor \
 
12
 
13
  RUN pip install --no-cache-dir -r requirements.txt
14
 
15
+ RUN pip install --no-cache-dir torch
16
 
17
  # Install supervisor to manage multiple processes
18
  RUN apt-get update && apt-get install -y supervisor \
requirements.txt CHANGED
@@ -9,3 +9,4 @@ accelerate
9
  huggingface_hub
10
  bitsandbytes
11
  peft
 
 
9
  huggingface_hub
10
  bitsandbytes
11
  peft
12
+ transformers==4.37.2