gemma can now gen with flash attn (revert transformers to 4.51.3)

This commit is contained in:
51616 2025-06-25 23:16:36 +09:00
parent 816377b4ac
commit 63f02fb4a7
2 changed files with 5 additions and 5 deletions

View file

@ -6,7 +6,7 @@ description = ""
readme = "README.md"
requires-python = ">= 3.10"
dependencies = [
"transformers==4.52.4",
"transformers==4.51.3",
"deepspeed==0.17.1",
"accelerate==1.6.0",
"datasets==3.6.0",