Removed safety checker

Author: djt

README.md

@@ -3,6 +3,8 @@
 
 This repo gives a basic framework for serving Stable Diffusion 1.5 in production using simple HTTP servers.
 
+*THIS MODEL HAS THE SAFETY CHECKER DISABLED*
+
 # Quickstart
 
 Deploy this model via the one-click template [here]((https://app.banana.dev/templates/djt/stable-diffusion-1.5).

app.py

@@ -10,7 +10,7 @@ def init():
     HF_AUTH_TOKEN = os.getenv("HF_AUTH_TOKEN")
     repo = 'runwayml/stable-diffusion-v1-5'
     scheduler = DPMSolverMultistepScheduler.from_pretrained(repo, subfolder="scheduler")
-    model = DiffusionPipeline.from_pretrained(repo, torch_dtype=torch.float16, revision="fp16", scheduler=scheduler, use_auth_token=HF_AUTH_TOKEN).to("cuda")    
+    model = DiffusionPipeline.from_pretrained(repo, torch_dtype=torch.float16, revision="fp16", scheduler=scheduler, use_auth_token=HF_AUTH_TOKEN, safety_checker=None).to("cuda")    
 
 def inference(model_inputs:dict):
     global model

demo.py

@@ -6,7 +6,7 @@ def download_model():
     HF_AUTH_TOKEN = os.getenv("HF_AUTH_TOKEN")
     repo = 'runwayml/stable-diffusion-v1-5'
     scheduler = DPMSolverMultistepScheduler.from_pretrained(repo, subfolder="scheduler")
-    model = DiffusionPipeline.from_pretrained(repo, torch_dtype=torch.float16, revision="fp16", scheduler=scheduler, use_auth_token=HF_AUTH_TOKEN) 
+    model = DiffusionPipeline.from_pretrained(repo, torch_dtype=torch.float16, revision="fp16", scheduler=scheduler, use_auth_token=HF_AUTH_TOKEN, safety_checker=None) 
 
 if __name__ == "__main__":
     download_model()