From 696f73c30d254191dacf6eb8ea75b938fdd44a76 Mon Sep 17 00:00:00 2001
From: Jaret Burkett <jaretburkett@gmail.com>
Date: Fri, 14 Jun 2024 17:09:47 -0600
Subject: [PATCH] Removed variant

---
 toolkit/stable_diffusion_model.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/toolkit/stable_diffusion_model.py b/toolkit/stable_diffusion_model.py
index 1f62f34..07e6bfd 100644
--- a/toolkit/stable_diffusion_model.py
+++ b/toolkit/stable_diffusion_model.py
@@ -201,7 +201,7 @@ class StableDiffusion:
                     model_path,
                     dtype=dtype,
                     device=self.device_torch,
-                    variant="fp16",
+                    # variant="fp16",
                     use_safetensors=True,
                     **load_args
                 )
@@ -389,7 +389,7 @@ class StableDiffusion:
                     load_safety_checker=False,
                     requires_safety_checker=False,
                     safety_checker=None,
-                    variant="fp16",
+                    # variant="fp16",
                     trust_remote_code=True,
                     **load_args
                 ).to(self.device_torch)
@@ -473,7 +473,7 @@ class StableDiffusion:
                     model_path,
                     dtype=dtype,
                     device=self.device_torch,
-                    variant="fp16",
+                    # variant="fp16",
                     use_safetensors=True,
                 ).to(self.device_torch)
             else:
@@ -1696,7 +1696,7 @@ class StableDiffusion:
                 model_path,
                 dtype=dtype,
                 device='cpu',
-                variant="fp16",
+                # variant="fp16",
                 use_safetensors=True,
             )
         else: