tests/kit/model_zoo/transformers/gptj.py · 21aa5de00b6138c019fae5f58024f2aff6f97a3a · OpenDAS / ColossalAI · GitLab

Find file Blame History Permalink

[gemini] hotfix NaN loss while using Gemini + tensor_parallel (#5150) · 21aa5de0
flybird11111 authored Dec 08, 2023
```
* fix

aaa

fix

fix

fix

* fix

* fix

* test ci

* fix ci

fix
```
21aa5de0

gptj.py 3.46 KB