Spaces:

Leteint
/

A1111

Running

Leteint commited on Jan 14

Commit

5c17ba6

verified ·

1 Parent(s): 0efbbe1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,8 @@ print("Chargement pipeline SDXL...")
 pipe = AutoPipelineForText2Image.from_pretrained(
     BASE_MODEL_ID,
     torch_dtype=DTYPE,
-    safety_checker=None,
     variant="fp16" if DTYPE == torch.float16 else None,
 )
 pipe.to(DEVICE)

 pipe = AutoPipelineForText2Image.from_pretrained(
     BASE_MODEL_ID,
     torch_dtype=DTYPE,
+    safety_checker=None,
+    requires_safety_checker=False,
     variant="fp16" if DTYPE == torch.float16 else None,
 )
 pipe.to(DEVICE)