fix: meta tensor device

vision_transformer_base.py

@@ -270,7 +270,7 @@ class DinoVisionTransformerBase(nn.Module):
         if drop_path_uniform is True:
             dpr = [drop_path_rate] * depth
         else:
-            dpr = [x.item() for x in torch.linspace(0, drop_path_rate, depth)]  # stochastic depth decay rule
+            dpr = torch.linspace(0, drop_path_rate, depth, device="cpu").tolist()  # stochastic depth decay rule
 
         if ffn_layer == DinoVisionTransformerFFNLayer.MLP:
             self.logger.info("Using MLP layer as FFN")