Consistent load behavior

developmentseed · Mar 15, 2024 · c079225 · c079225
1 parent 50c22f9
commit c079225
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/services/gpu/lib/ModelSrv.py b/services/gpu/lib/ModelSrv.py
@@ -720,8 +720,7 @@ def meta_save_checkpoint(self, body, websocket):
                 )
             )
 
-        self.chk = checkpoint
-        return checkpoint
+        self.chk = self.api.get_checkpoint(checkpoint["id"])
 
 
 def done_processing(modelsrv):