allenai / allenai/unified-io-2.pytorch
Minor suggestion for pytorch version requirement of UIO2
Offen
- 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.