Open wangdalu4399 opened 7 months ago
x4_prompt = self.prompt3(x4, prompt_param3) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!
您好,您解决了么
self.transformer_block = nn.Sequential(*[
TransformerBlock(dim=dim // num_splits, num_heads=1, ffn_expansion_factor=2.66, bias=False,
LayerNorm_type='WithBias') for _ in range(num_splits)])
添加nn.Sequential(*)
x4_prompt = self.prompt3(x4, prompt_param3) ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ RuntimeError: Expected all tensors to be on the same device, but found at least two devices, cuda:0 and cpu!