text-generation-inference/integration-tests
Nicolas Patry 1b434e8019 Refactor dead code - Removing all flash_xxx.py files. (#2166)
* Refactor dead code.

* First working step.

* Remove a lot of duplicated code.

* More dead code.

* More cleanup.

* Fix Santacoder test.

* Fixing the simple tests.

* Fixing sharding.

* Fixes for VLM.

* Fixing santacoder (num_kv_heads hardcoded).

* Removing more dead code.

* Fixing `config.n_head`.

* Stopping earlier because of `<end_of_utterance>` in idefics2.

* Addresses comments.

* Removing the dead code.

* Fuse back mistral into FlashCausalLM.

* Finish removal.

* Fixing docs + causal_lm `batch_class`.

* Fixing docs + causal.lm.

* Add default to Gemma Causality.

* Default value for gemma/gemma2.

* Wrong default.
2024-09-25 05:20:28 +00:00
..
images Pali gemma modeling (#1895) 2024-07-17 05:36:58 +00:00
models Refactor dead code - Removing all flash_xxx.py files. (#2166) 2024-09-25 05:20:28 +00:00
conftest.py Add pytest release marker (#2114) 2024-09-24 03:52:50 +00:00
poetry.lock fix: improve tool type, bump pydantic and outlines (#1650) 2024-04-25 12:34:55 +03:00
pyproject.toml v2.0.1 2024-06-03 15:39:47 +03:00
pytest.ini chore: add pre-commit (#1569) 2024-04-24 15:32:02 +03:00
requirements.txt fix: improve tool type, bump pydantic and outlines (#1650) 2024-04-25 12:34:55 +03:00