Skip to content

Commit

Permalink
chore: Update Python Pip libraries (#412)
Browse files Browse the repository at this point in the history
**Reason for Change**:
Dependabot notice, update libraries
  • Loading branch information
ishaansehgal99 authored May 16, 2024
1 parent 63fb951 commit 8587d35
Show file tree
Hide file tree
Showing 3 changed files with 31 additions and 20 deletions.
4 changes: 2 additions & 2 deletions presets/inference/text-generation/inference_api.py
Original file line number Diff line number Diff line change
Expand Up @@ -175,8 +175,8 @@ class GenerateKwargs(BaseModel):
pad_token_id: Optional[int] = tokenizer.pad_token_id
eos_token_id: Optional[int] = tokenizer.eos_token_id
class Config:
extra = Extra.allow # Allows for additional fields not explicitly defined
schema_extra = {
extra = 'allow' # Allows for additional fields not explicitly defined
json_schema_extra = {
"example": {
"max_length": 200,
"temperature": 0.7,
Expand Down
26 changes: 16 additions & 10 deletions presets/inference/text-generation/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,14 +1,20 @@
# Dependencies for TFS
transformers==4.37.2

# Core Dependencies
transformers==4.40.2
torch==2.2.0
accelerate==0.23.0
fastapi==0.109.1
pydantic==1.10.9
uvicorn[standard]==0.23.2
accelerate==0.30.1
fastapi>=0.111.0,<0.112.0 # Allow patch updates
pydantic>=2.7.1,<2.8 # Allow patch updates
uvicorn[standard]>=0.29.0,<0.30.0 # Allow patch updates

# Utility libraries
bitsandbytes==0.42.0
deepspeed==0.11.1
gputil==1.4.0
psutil==5.9.8

# Less critical, can be latest
deepspeed
gputil
psutil
# For UTs
pytest==8.0.0
httpx==0.26.0
pytest
httpx
21 changes: 13 additions & 8 deletions presets/tuning/tfs/requirements.txt
Original file line number Diff line number Diff line change
@@ -1,14 +1,19 @@
# Core Dependencies
transformers==4.40.2
torch==2.2.0
accelerate==0.30.1
fastapi>=0.111.0,<0.112.0 # Allow patch updates
pydantic>=2.7.1,<2.8 # Allow patch updates
uvicorn[standard]>=0.29.0,<0.30.0 # Allow patch updates

# Utility libraries
datasets==2.16.1
peft==0.8.2
transformers==4.38.2
torch==2.2.0
accelerate==0.27.2
fastapi==0.109.1
pydantic==1.10.9
uvicorn[standard]==0.23.2
bitsandbytes==0.42.0
gputil==1.4.0

# Less critical, can be latest
deepspeed
loralib
einops
xformers
deepspeed
gputil

0 comments on commit 8587d35

Please sign in to comment.