aljaz/generative-models
1
0
Fork 0
mirror of https://github.com/Stability-AI/generative-models.git synced 2026-01-09 08:34:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #28 from jenuk/fix-samples_z

Browse Source 
Fix `samples_z` undefined
This commit is contained in:
Tim Dockhorn
2023-07-09 10:35:04 -07:00
committed by GitHub
parent 9d5ace911e 7b1978e055
commit ea89ce793d
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
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,