TensorRT-LLMs/examples/models/core/recurrentgemma/requirements.txt
mpikulski 93a4b7f1b6
[None][chore] update torch_dtype -> dtype in 'transformers' (#8263)
Signed-off-by: ixlmar <206748156+ixlmar@users.noreply.github.com>
2025-10-15 17:09:30 +09:00

13 lines
258 B
Plaintext

-c ../../../constraints.txt
tensorrt_llm>=0.0.0.dev0
git+https://github.com/google-deepmind/recurrentgemma.git@8a32e365
flax>=0.8.2
jax~=0.4.23
orbax-checkpoint==0.5.7
transformers>=4.56.0
datasets==3.1.0
evaluate
rouge_score
sentencepiece
jaxtyping<=0.2.34