Skip to content

Commit a137e17

Browse files
committed
add tiktoken to test dependencies
1 parent b4c9fde commit a137e17

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

setup.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -130,6 +130,7 @@
130130
"regex!=2019.12.17",
131131
"requests",
132132
"tensorboard",
133+
"tiktoken>=0.7.0",
133134
"torch>=1.4",
134135
"torchvision",
135136
"transformers>=4.41.2",
@@ -226,6 +227,7 @@ def run(self):
226227
"safetensors",
227228
"sentencepiece",
228229
"scipy",
230+
"tiktoken",
229231
"torchvision",
230232
"transformers",
231233
"phonemizer",

src/diffusers/dependency_versions_table.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,6 +38,7 @@
3838
"regex": "regex!=2019.12.17",
3939
"requests": "requests",
4040
"tensorboard": "tensorboard",
41+
"tiktoken": "tiktoken>=0.7.0",
4142
"torch": "torch>=1.4",
4243
"torchvision": "torchvision",
4344
"transformers": "transformers>=4.41.2",

0 commit comments

Comments
 (0)