Upload modeling_gluformer.py

modeling_gluformer.py

@@ -356,5 +356,12 @@ class GluformerForTimeSeries(PreTrainedModel):
             timestamps = timestamps.unsqueeze(0)
             glucose_values = glucose_values.unsqueeze(0)
         x_id, x_enc, x_dec, x_mark_enc, y_mark_dec = self.preprocessor(subject_id, timestamps, glucose_values)
-        output, log_var = self.model(x_id, x_enc.to(self.device), x_mark_enc.to(self.device), x_dec.to(self.device), y_mark_dec.to(self.device))
+        if self.device is not None:
+            x_id = x_id.to(self.device)
+            x_enc = x_enc.to(self.device)
+            x_dec = x_dec.to(self.device)
+            x_mark_enc = x_mark_enc.to(self.device)
+            y_mark_dec = y_mark_dec.to(self.device)
+            self.model.to(self.device)
+        output, log_var = self.model(x_id, x_enc, x_mark_enc, x_dec, y_mark_dec)
         return self.preprocessor.unnormalize_glucose(output).cpu(), log_var.cpu()