Jinghong Zeng
Jinghong Zeng
I met the same bug when running train_sft.sh in the applications-Chat-examples path. It seems that `import colossalai.zero.gemini` will lead to an importerror.
> @sharejing ,我之前弄错了colosaala的版本,官方的tag: 0.2.7的容器镜像里面的colosaalai的版本是0.2.0的,但我试用的代码例子是v0.2.7里面的。更新下容器里面的软件版本就行。因为最新的代码,ColoInitContext在这个位置: from colossalai.utils.model.colo_init_context import ColoInitContext @LarryZhangy hi, I didn't find ColoInitContext in colossalai.utils.model.colo_init_context from the main branch. wondering how you solved this problem?
> > > @sharejing ,我之前弄错了colosaala的版本,官方的tag: 0.2.7的容器镜像里面的colosaalai的版本是0.2.0的,但我试用的代码例子是v0.2.7里面的。更新下容器里面的软件版本就行。因为最新的代码,ColoInitContext在这个位置: from colossalai.utils.model.colo_init_context import ColoInitContext > > > > > > @LarryZhangy hi, I didn't find ColoInitContext in colossalai.utils.model.colo_init_context from the main branch. wondering how...
@sharejing I figured out this importerror on my device. This is due to difference between colossaial packages in the python directory and those git from source. My steps to solve...