amazonaws-la commited on
Commit
f3eaa57
·
verified ·
1 Parent(s): b70be97

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -65,7 +65,7 @@ def generate(
65
  if torch.cuda.is_available():
66
 
67
  if not use_vae:
68
- pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model, torch_dtype=torch.float16, safety_checker=StableDiffusionSafetyChecker)
69
 
70
  if use_vae:
71
  vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)
 
65
  if torch.cuda.is_available():
66
 
67
  if not use_vae:
68
+ pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model, torch_dtype=torch.float16, requires_safety_checker=True)
69
 
70
  if use_vae:
71
  vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)