pixtral
Showing
- pyproject.toml 2 additions, 1 deletionpyproject.toml
- src/mistral_inference/args.py 17 additions, 2 deletionssrc/mistral_inference/args.py
- src/mistral_inference/generate.py 13 additions, 0 deletionssrc/mistral_inference/generate.py
- src/mistral_inference/main.py 60 additions, 8 deletionssrc/mistral_inference/main.py
- src/mistral_inference/rope.py 30 additions, 2 deletionssrc/mistral_inference/rope.py
- src/mistral_inference/transformer.py 65 additions, 140 deletionssrc/mistral_inference/transformer.py
- src/mistral_inference/transformer_utils.py 169 additions, 0 deletionssrc/mistral_inference/transformer_utils.py
- src/mistral_inference/vision_encoder.py 144 additions, 0 deletionssrc/mistral_inference/vision_encoder.py
- tests/test_generate.py 72 additions, 5 deletionstests/test_generate.py
Loading
Please register or sign in to comment