Bugfix to Partial diffusion where partial_t doesn't affect output (#138)

Return noise_schedule instead of t_hat

models/rfd3/src/rfd3/model/inference_sampler.py

@@ -120,8 +120,10 @@ class SampleDiffusionWithMotif(SampleDiffusionConfig):
                 ranked_logger.info(
                     f"Using fallback: final step with t={noise_schedule[0].item():.6f}"
                 )
+        else:
+            noise_schedule = t_hat
 
-        return t_hat
+        return noise_schedule
 
     def _get_initial_structure(
         self,