This commit is contained in:
yumoqing 2025-06-19 17:31:10 +08:00
parent 58c4af5c8b
commit 4b859e235b
5 changed files with 2 additions and 0 deletions

View File

@ -9,6 +9,8 @@ license = {text = "MIT"}
dependencies = [
"torch",
"transformers",
"sentence-transformers>=2.7.0",
# "flash_attention_2",
"mistral-common",
"accelerate"
]