aljaz/generative-models
1
0
Fork 0
mirror of https://github.com/Stability-AI/generative-models.git synced 2026-02-02 04:14:27 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix missing index

Browse Source
This commit is contained in:
Stephan Auerhahn
2023-08-10 12:24:12 -07:00
parent 3e7ada70c5
commit 26b10f56f3
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/demo/streamlit_helpers.py
View File

@@ -47,6 +47,8 @@ def init_st(spec: SamplingSpec, load_ckpt=True, load_filter=True) -> Dict[str, A
state["params"] = SamplingParams()
if load_filter:
state["filter"] = DeepFloydDataFiltering(verbose=False)
else:
state["filter"] = None
return state