[Bugfix][Model] Refactor OLMo model to support new HF format in transformers 4.40.0 (#4324)

Co-authored-by: Woosuk Kwon <woosuk.kwon@berkeley.edu>
This commit is contained in:
Isotr0py
2024-04-26 00:35:56 +08:00
committed by GitHub
parent 479d69fad0
commit fbf152d976
5 changed files with 150 additions and 162 deletions

View File

@@ -26,7 +26,6 @@ requests
ray
peft
awscli
ai2-olmo # required for OLMo
# Benchmarking
aiohttp