Added cosmos3 model (#46146)
* Added cosmos3 model
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* bugfix for processing_utils.py
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* Add extra tests
* Rename architecture; linter fixes
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* Rename the architecture again for now
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* Adapted to new Cosmos3 checkpoint format
* Improve the checkpoint loading
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* Simplified docs
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* Renamed cosmos3 to cosmos3_reasoner
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* Removed cosmos3 tests
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* Revert processing fix
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* Answered review comments
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* Rename to Cosmos3ReasonerForConditionalGeneration
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* Renamed config key
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* Run make fix-repo
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* Refactored naming
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* Some more renaming; fix linter issues; Add Cosmos3 Processor
Signed-off-by: MaciejBalaNV <mbala@nvidia.com>
* Linter fixes
Signed-off-by: MaciejBalaNV <mbala@nvidia.com>
* Multiple fixes
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* fix test and linter
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* Make fix-repo fixes
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* Linter fix
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* Change date
Signed-off-by: Maciej Bala <mbala@nvidia.com>
* rename it all and load existing AutoModel
* docs
* fix repo
* fix docs
* ugly typo
* adjust expectation
* dont sample in test!
* forgot this one
* style
---------
Signed-off-by: Maciej Bala <mbala@nvidia.com>
Signed-off-by: MaciejBalaNV <mbala@nvidia.com>
Co-authored-by: raushan <raushan@huggingface.co>
