Commit
·
dc90bb8
1
Parent(s):
f33e28a
updated poetry
Browse files- pyproject.toml +2 -0
pyproject.toml
CHANGED
|
@@ -6,8 +6,10 @@ authors = ["Your Name <[email protected]>"]
|
|
| 6 |
|
| 7 |
[tool.poetry.dependencies]
|
| 8 |
python = "^3.10"
|
|
|
|
| 9 |
clip = {git = "https://github.com/openai/CLIP.git", rev = "main"}
|
| 10 |
torchvision = "0.19"
|
|
|
|
| 11 |
vllm = "^0.6.3"
|
| 12 |
pillow = "10.3.0"
|
| 13 |
PyMuPDF = "^1.24.13"
|
|
|
|
| 6 |
|
| 7 |
[tool.poetry.dependencies]
|
| 8 |
python = "^3.10"
|
| 9 |
+
torch = "^2.4.0"
|
| 10 |
clip = {git = "https://github.com/openai/CLIP.git", rev = "main"}
|
| 11 |
torchvision = "0.19"
|
| 12 |
+
flash-attn = "^2.7.0.post2"
|
| 13 |
vllm = "^0.6.3"
|
| 14 |
pillow = "10.3.0"
|
| 15 |
PyMuPDF = "^1.24.13"
|