Spaces:

Prgckwb
/

stable-diffusion-demo

Sleeping

Prgckwb commited on Jun 17

Commit

5566482

•

1 Parent(s): 255b6b2

:memo: Change

Files changed (1) hide show

app.py CHANGED Viewed

@@ -97,7 +97,7 @@ def inference(
     current_model_id = model_id
     if not safety_checker:
-        pipe.use_safety_checker = None
     if model_id not in DIFFUSERS_MODEL_IDS:
         progress(0.3, 'Loading Textual Inversion...')

     current_model_id = model_id
     if not safety_checker:
+        pipe.safety_checker = None
     if model_id not in DIFFUSERS_MODEL_IDS:
         progress(0.3, 'Loading Textual Inversion...')