aljaz/generative-models
1
0
Fork 0
mirror of https://github.com/Stability-AI/generative-models.git synced 2025-12-19 22:34:22 +01:00
Code Issues Packages Projects Releases Wiki Activity

Only do refiner step if samples are actually available

Browse Source
This commit is contained in:
jenuk
2023-07-07 07:48:21 +00:00
parent 9d5ace911e
commit 7b1978e055
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
scripts/demo/sampling.py
View File

@@ -311,8 +311,9 @@ if __name__ == "__main__":
samples, samples_z = out samples, samples_z = out
else: else:
samples = out samples = out
samples_z = None
if add_pipeline: if add_pipeline and samples_z is not None:
st.write("**Running Refinement Stage**") st.write("**Running Refinement Stage**")
samples = apply_refiner( samples = apply_refiner(
samples_z, samples_z,