Only do refiner step if samples are actually available

LinearFalcon · Jul 6, 2024 · 3a4e63e · 3a4e63e
1 parent 94c146d
commit 3a4e63e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/scripts/demo/sampling.py b/scripts/demo/sampling.py
@@ -311,8 +311,9 @@ def apply_refiner(
         samples, samples_z = out
     else:
         samples = out
+        samples_z = None
 
-    if add_pipeline:
+    if add_pipeline and samples_z is not None:
         st.write("**Running Refinement Stage**")
         samples = apply_refiner(
             samples_z,