allenai / allenai/OLMoASR

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

Offen
#56 1 Kommentar 1 Reaktion 0 zugewiesene Personen Auf GitHub ansehen
Vorherrschende Sprache
Python
Sterne
493
Forks
48
PR-Merge-Kennzahlen
Keine gemergten PRs in 30 T.

Beschreibung

```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'
```

Beitragsleitfaden

Für dieses Repository ist kein Beitragsleitfaden indexiert

Bewertung

Dieses Issue wurde noch nicht bewertet.

Neue Issues direkt in Ihr Postfach

Eine kurze Übersicht über anfängerfreundliche GitHub-Issues.