Sameen

Same problem, changed wsl2 memory to 16g still not work.

> Changing torch.float16 to torch.float32 might solve the problem I think it worked

> 傻屌 咋还用中文回复呢，抓紧学学用你洋爹的语言回复啊。 不然滚出去了被人骂了还得陪笑不是么