ux(progress): add finer init progress steps and use weights_only for checkpoint loading

app_wip.py

@@ -95,7 +95,11 @@ def initialize_pipeline(progress: gr.Progress | None = None):
 
     if progress is not None:
         progress(0.11, desc="Init: loading reward forcing checkpoint")
-    state_dict = torch.load(CHECKPOINT_PATH, map_location="cpu")
+    state_dict = torch.load(
+        CHECKPOINT_PATH,
+        map_location="cpu",
+        weights_only=True,
+    )
     pipeline.generator.load_state_dict(state_dict)
 
     checkpoint_step = os.path.basename(os.path.dirname(CHECKPOINT_PATH))