Update modeling_ced.py

modeling_ced.py

@@ -367,7 +367,7 @@ class CedModel(CedPreTrainedModel):
         self.freq_pos_embed = nn.Parameter(torch.randn(1, config.embed_dim, self.patch_embed.grid_size[0], 1) * 0.02)
         norm_layer = partial(nn.LayerNorm, eps=1e-6)
         act_layer = nn.GELU
-        dpr = [x.item() for x in torch.linspace(0, config.drop_path_rate, config.depth)]  # stochastic depth decay rule
+        dpr = [x.item() for x in torch.linspace(0, config.drop_path_rate, config.depth, device="cpu")]  # stochastic depth decay rule
         self.pos_drop = nn.Dropout(p=config.drop_rate)
         self.blocks = nn.Sequential(
             *[