Spaces:

Divyanshu04
/

Text2image-streamlit

Runtime error

Divyanshu04 commited on Nov 24, 2023

Commit

7a7d572

1 Parent(s): 2d29779

Changes

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ login(token  = "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")
 # headers = {"Authorization": "Bearer xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}
 # pipe = StableDiffusionPipeline.from_pretrained("Divyanshu04/Finetuned-sd-vae", safety_checker=None, torch_dtype=torch.float32).to("cuda")
-pipe = StableDiffusionPipeline.from_pretrained("Divyanshu04/Finetuned-sd-vae", torch_dtype=torch.float16,  use_safetensors=True, use_auth_token = "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")
 pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
 pipe.enable_xformers_memory_efficient_attention()
 g_cuda = None

 # headers = {"Authorization": "Bearer xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx"}
 # pipe = StableDiffusionPipeline.from_pretrained("Divyanshu04/Finetuned-sd-vae", safety_checker=None, torch_dtype=torch.float32).to("cuda")
+pipe = StableDiffusionPipeline.from_pretrained("Divyanshu04/Finetuned-model", torch_dtype=torch.float16,  use_safetensors=True, use_auth_token = "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")
 pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)
 pipe.enable_xformers_memory_efficient_attention()
 g_cuda = None