Upload modeling_gluformer.py

modeling_gluformer.py

@@ -259,7 +259,7 @@ class Gluformer(nn.Module):
         dec_out = self.dec_embedding(x_id, x_dec, x_mark_dec)
         dec_out = self.decoder(dec_out, enc_out)
         dec_out = self.projection(dec_out)
-        return dec_out[:, -self.len_pred:, :], var_out 
+        return dec_out[:, -self.len_pred:, :], var_out
 
 class GluformerConfig(PretrainedConfig):
     model_type = "gluformer"
@@ -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)
+        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), log_var
+        return self.preprocessor.unnormalize_glucose(output).cpu(), log_var.cpu()