fix missing index

Stability-AI · Aug 10, 2023 · 26b10f5 · 26b10f5
1 parent 3e7ada7
commit 26b10f5
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/scripts/demo/streamlit_helpers.py b/scripts/demo/streamlit_helpers.py
@@ -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