github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-12 20:54:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

[example] add palm pytorch version (#2172)

Browse Source
This commit is contained in:
Jiarui Fang
2022-12-22 10:15:34 +08:00
committed by GitHub
parent 12e7bcd720
commit 27327a4c90
7 changed files with 454 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_gemini/update/test_convert_torch_module.py
View File

@@ -1,6 +1,8 @@
import os
from functools import partial
import pytest
import torch
import torch.multiprocessing as mp
import colossalai