add model

model_segvol_single.py

@@ -101,7 +101,7 @@ class SegVolModel(PreTrainedModel):
         ## inference
         with torch.no_grad():
             logits_single_cropped = sliding_window_inference(
-                    image_single_cropped.cuda(), prompt_reflection,
+                    image_single_cropped.to(self.custom_device), prompt_reflection,
                     self.config.spatial_size, 1, self.model, 0.5,
                     text=text_prompt,
                     use_box=bbox_prompt is not None,