Skip to content

Commit 8048683

Browse files
committed
adds PEFT req. for lora scaling export issue and fix diffusers
1 parent 7b861a7 commit 8048683

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

models/requirements.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,7 @@ shark-turbine @ git+https://github.com/iree-org/iree-turbine.git@main
44
transformers==4.37.1
55
torchsde
66
accelerate
7+
peft
78
diffusers @ git+https://github.com/nod-ai/diffusers@0.29.0.dev0-shark
89
brevitas @ git+https://github.com/Xilinx/brevitas.git@6695e8df7f6a2c7715b9ed69c4b78157376bb60b
910
# turbine tank downloading/uploading

models/setup.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def load_version_info():
5757
"Shark-Turbine",
5858
"protobuf",
5959
"sentencepiece",
60-
"transformers==4.37.1",
60+
"transformers>=4.37.1",
6161
"accelerate",
6262
"diffusers==0.29.0.dev0",
6363
"azure-storage-blob",

0 commit comments

Comments
 (0)