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",

8
uv.lock generated
View file

@ -728,7 +728,7 @@ requires-dist = [
{ name = "tensorboard" },
{ name = "tensorboardx" },
{ name = "torchmetrics" },
{ name = "transformers", specifier = "==4.52.4" },
{ name = "transformers", specifier = "==4.51.3" },
{ name = "vllm", specifier = "==0.8.5" },
{ name = "wandb" },
]
@ -5036,7 +5036,7 @@ wheels = [
[[package]]
name = "transformers"
version = "4.52.4"
version = "4.51.3"
source = { registry = "https://pypi.org/simple" }
dependencies = [
{ name = "filelock" },
@ -5050,9 +5050,9 @@ dependencies = [
{ name = "tokenizers" },
{ name = "tqdm" },
]
sdist = { url = "https://files.pythonhosted.org/packages/da/a9/275037087f9d846580b02f2d7cae0e0a6955d46f84583d0151d6227bd416/transformers-4.52.4.tar.gz", hash = "sha256:aff3764441c1adc192a08dba49740d3cbbcb72d850586075aed6bd89b98203e6", size = 8945376, upload-time = "2025-05-30T09:17:17.947Z" }
sdist = { url = "https://files.pythonhosted.org/packages/f1/11/7414d5bc07690002ce4d7553602107bf969af85144bbd02830f9fb471236/transformers-4.51.3.tar.gz", hash = "sha256:e292fcab3990c6defe6328f0f7d2004283ca81a7a07b2de9a46d67fd81ea1409", size = 8941266, upload-time = "2025-04-14T08:15:00.485Z" }
wheels = [
{ url = "https://files.pythonhosted.org/packages/96/f2/25b27b396af03d5b64e61976b14f7209e2939e9e806c10749b6d277c273e/transformers-4.52.4-py3-none-any.whl", hash = "sha256:203f5c19416d5877e36e88633943761719538a25d9775977a24fe77a1e5adfc7", size = 10460375, upload-time = "2025-05-30T09:17:14.477Z" },
{ url = "https://files.pythonhosted.org/packages/a9/b6/5257d04ae327b44db31f15cce39e6020cc986333c715660b1315a9724d82/transformers-4.51.3-py3-none-any.whl", hash = "sha256:fd3279633ceb2b777013234bbf0b4f5c2d23c4626b05497691f00cfda55e8a83", size = 10383940, upload-time = "2025-04-14T08:13:43.023Z" },
]
[[package]]