allenai / allenai/OLMoASR

large-v2 will not load, returns `KeyError: 'dims'`

Aperta
#56 1 commento 1 reazione 0 assegnatari Vedi su GitHub
Lingua principale
Python
Stelle
493
Fork
48
Metriche di merge delle PR
Nessuna PR unita negli ultimi 30g

Descrizione

```shell
python -c "import olmoasr; olmoasr.load_model('large-v2', inference=True)"

Downloading large-v2 model from https://huggingface.co/allenai/OLMoASR/resolve/main/models/OLMoASR-large.en-v2.pt...
Saving to: .cache/olmoasr/OLMoASR-large-v2.pt
Downloading... 100%
Model downloaded successfully: .cache/olmoasr/OLMoASR-large-v2.pt
Traceback (most recent call last):
File "", line 1, in
File ".venv/lib/python3.10/site-packages/olmoasr/__init__.py", line 156, in load_model
dims = ModelDimensions(**checkpoint["dims"])
KeyError: 'dims'
```

Guida per i contributori

Nessuna guida per i contributori indicizzata per questo repository

Valutazione

Questa issue non è ancora stata valutata.

Ricevi le nuove issue nella tua casella

Un breve riepilogo di issue GitHub adatte ai principianti.