Argonne-1.5 / model.py

Commit History

Fix generate method to handle CausalLMOutput, plus other updates
ea46d13

Youzhi Yu commited on

Update ArgonneModel generate method to accept standard HF params
5324bdd

Youzhi Yu commited on

Initial commit of Argonne-1.5 model
4395cf9

Youzhi Yu commited on