mirror of
https://github.com/SakanaAI/doc-to-lora.git
synced 2026-07-26 17:11:02 +02:00
gemma can now gen with flash attn (revert transformers to 4.51.3)
This commit is contained in:
parent
816377b4ac
commit
63f02fb4a7
2 changed files with 5 additions and 5 deletions
|
|
@ -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",
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue