allenai / allenai/unified-io-2.pytorch

Minor suggestion for pytorch version requirement of UIO2

Offen
#11 0 Kommentare 0 Reaktionen 0 zugewiesene Personen Auf GitHub ansehen
Vorherrschende Sprache
Python
Sterne
78
Forks
8
PR-Merge-Kennzahlen
Keine gemergten PRs in 30 T.

Beschreibung

Since **nn.LayerNorm()** in **pytorch <2.1.0** [seems](https://pytorch.org/docs/2.0/_modules/torch/nn/modules/normalization.html#LayerNorm) to exist without keyword argument `'bias'`,
following error occurs while loading pretrained ckpt of UIO2 model:
```
TypeError: __init__() got an unexpected keyword argument 'bias'
```

Therefore, I recommend whom are interested in testing the pretrained version of UIO2 to try **pytorch >= 2.1.0**.

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.