Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ login(token=a )
|
|
13 |
|
14 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
15 |
#model_repo_id = "stabilityai/sdxl-turbo" # Replace to the model you would like to use
|
16 |
-
|
17 |
|
18 |
|
19 |
|
@@ -58,7 +58,7 @@ var_2="nota-ai/bk-sdm-small"
|
|
58 |
|
59 |
|
60 |
pipe = StableDiffusionPipeline.from_pretrained(
|
61 |
-
|
62 |
#pipe.vae = AutoencoderTiny.from_pretrained(
|
63 |
# "sayakpaul/taesd-diffusers", torch_dtype=torch_dtype, use_safetensors=True)
|
64 |
pipe = pipe.to(device)
|
|
|
13 |
|
14 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
15 |
#model_repo_id = "stabilityai/sdxl-turbo" # Replace to the model you would like to use
|
16 |
+
model_repo_id = "stable-diffusion-v1-5/stable-diffusion-v1-5"
|
17 |
|
18 |
|
19 |
|
|
|
58 |
|
59 |
|
60 |
pipe = StableDiffusionPipeline.from_pretrained(
|
61 |
+
model_repo_id, torch_dtype=torch_dtype, use_safetensors=True)
|
62 |
#pipe.vae = AutoencoderTiny.from_pretrained(
|
63 |
# "sayakpaul/taesd-diffusers", torch_dtype=torch_dtype, use_safetensors=True)
|
64 |
pipe = pipe.to(device)
|