test

app.py

@@ -9,8 +9,8 @@ import streamlit as st
 
 # model_path = WEIGHTS_DIR             # If you want to use previously trained model saved in gdrive, replace this with the full path of model in gdrive
 
-pipe = StableDiffusionPipeline.from_pretrained("Divyanshu04/Finetuned-sd-vae", safety_checker=None, torch_dtype=torch.float32).to("cuda")
-# pipe = DiffusionPipeline.from_pretrained("Divyanshu04/Finetuned-sd-vae", torch_dtype=torch.float32)
+# pipe = StableDiffusionPipeline.from_pretrained("Divyanshu04/Finetuned-sd-vae", safety_checker=None, torch_dtype=torch.float32).to("cuda")
+pipe = DiffusionPipeline.from_pretrained("Divyanshu04/Finetuned-sd-vae", torch_dtype=torch.float32)
 pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
 pipe.enable_xformers_memory_efficient_attention()
 g_cuda = None