Skip to content

Commit 3a4e63e

Browse files
jenukLinearFalcon
authored andcommitted
Only do refiner step if samples are actually available
1 parent 94c146d commit 3a4e63e

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

scripts/demo/sampling.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -311,8 +311,9 @@ def apply_refiner(
311311
samples, samples_z = out
312312
else:
313313
samples = out
314+
samples_z = None
314315

315-
if add_pipeline:
316+
if add_pipeline and samples_z is not None:
316317
st.write("**Running Refinement Stage**")
317318
samples = apply_refiner(
318319
samples_z,

0 commit comments

Comments
 (0)