QishengL

Multiple GPU problems?

5

I use U2Dmodel with nn.DataParallel if torch.cuda.device_count() > 1: model = nn.DataParallel(model) Then I try to get the output. model_output = model(noisy_images, timesteps) The model _output is UNet2DOutput(sample=) model _output.sample...

Installation Problem

2

After following the Setup, I found that I need to downgrade protobuf to 3.20 and intall wrapt, opt_einsum, gast, astunparse, termcolor, keras-preprocessing, and python-flatbuffers to load the package and model...

预训练模型链接过期了能补一下吗

2

预训练模型链接过期了能补一下吗？感谢