sd 1.5 model

usamaehsan · Nov 22, 2023 · 47677b5 · 47677b5
1 parent e1b07d0
commit 47677b5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/predict.py b/predict.py
@@ -141,7 +141,7 @@ def setup(self):
 
         vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse")
         self.pipe = StableDiffusionPipeline.from_pretrained(
-            "SG161222/Realistic_Vision_V5.0_noVAE", torch_dtype=torch.float16,
+            "runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16,
             # local_files_only=True,
             vae= vae
         )