Revert "switch back to full model"

This reverts commit 4dccac5.
Hades32 · Sep 12, 2022 · bbae240 · bbae240
1 parent 4dccac5
commit bbae240
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 7 deletions.
diff --git a/app.py b/app.py
@@ -16,11 +16,9 @@ def init():
 
     model = StableDiffusionPipeline.from_pretrained(
         "CompVis/stable-diffusion-v1-4",
-    #    revision="fp16",
-    #    torch_dtype=torch.float16,
-        scheduler=lms,
-        use_auth_token=HF_AUTH_TOKEN
-    ).to("cuda")
+        revision="fp16",
+        torch_dtype=torch.float16,
+        scheduler=lms, use_auth_token=HF_AUTH_TOKEN).to("cuda")
 
 # Inference is ran for every server call
 # Reference your preloaded global model variable here.

diff --git a/download.py b/download.py
@@ -18,8 +18,8 @@ def download_model():
 
     model = StableDiffusionPipeline.from_pretrained(
         "CompVis/stable-diffusion-v1-4", 
-        #revision="fp16",
-        #torch_dtype=torch.float16,
+        revision="fp16",
+        torch_dtype=torch.float16,
         scheduler=lms,
         use_auth_token=HF_AUTH_TOKEN
     )