Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -23,7 +23,7 @@ else:
|
|
23 |
torch_dtype = torch.float32
|
24 |
"""
|
25 |
pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
|
26 |
-
pipe = pipe.to(device)
|
27 |
"""
|
28 |
|
29 |
|
@@ -60,7 +60,7 @@ tiling, trimmed, unfocused, unattractive, unnatural pose, unreal engine, unsight
|
|
60 |
|
61 |
|
62 |
pipe = StableDiffusionPipeline.from_pretrained(
|
63 |
-
"
|
64 |
#pipe.vae = AutoencoderTiny.from_pretrained(
|
65 |
# "sayakpaul/taesd-diffusers", torch_dtype=torch_dtype, use_safetensors=True)
|
66 |
pipe = pipe.to(device)
|
|
|
23 |
torch_dtype = torch.float32
|
24 |
"""
|
25 |
pipe = DiffusionPipeline.from_pretrained(model_repo_id, torch_dtype=torch_dtype)
|
26 |
+
pipe = pipe.to(device) ###### это потом если что удалить "nota-ai/bk-sdm-small",
|
27 |
"""
|
28 |
|
29 |
|
|
|
60 |
|
61 |
|
62 |
pipe = StableDiffusionPipeline.from_pretrained(
|
63 |
+
"madebyollin/taesdxl", torch_dtype=torch_dtype, use_safetensors=True)
|
64 |
#pipe.vae = AutoencoderTiny.from_pretrained(
|
65 |
# "sayakpaul/taesd-diffusers", torch_dtype=torch_dtype, use_safetensors=True)
|
66 |
pipe = pipe.to(device)
|