Spaces:
Runtime error
Runtime error
Hotfix lpw not working
Browse filesUnintentionally disabled `from_single_file` because I'm stupid. Will ask
@DamarJati
to fix it later.
app.py
CHANGED
|
@@ -30,7 +30,7 @@ OUTPUT_DIR = os.getenv("OUTPUT_DIR", "./outputs")
|
|
| 30 |
|
| 31 |
MODEL = os.getenv(
|
| 32 |
"MODEL",
|
| 33 |
-
"
|
| 34 |
)
|
| 35 |
|
| 36 |
torch.backends.cudnn.deterministic = True
|
|
@@ -44,19 +44,13 @@ def load_pipeline(model_name):
|
|
| 44 |
"madebyollin/sdxl-vae-fp16-fix",
|
| 45 |
torch_dtype=torch.float16,
|
| 46 |
)
|
| 47 |
-
pipeline =
|
| 48 |
-
StableDiffusionXLPipeline.from_single_file
|
| 49 |
-
if MODEL.endswith(".safetensors")
|
| 50 |
-
else StableDiffusionXLPipeline.from_pretrained
|
| 51 |
-
)
|
| 52 |
|
| 53 |
pipe = pipeline(
|
| 54 |
model_name,
|
| 55 |
vae=vae,
|
| 56 |
torch_dtype=torch.float16,
|
| 57 |
custom_pipeline="lpw_stable_diffusion_xl",
|
| 58 |
-
use_safetensors=True,
|
| 59 |
-
add_watermarker=False,
|
| 60 |
use_auth_token=HF_TOKEN,
|
| 61 |
)
|
| 62 |
|
|
@@ -392,4 +386,4 @@ with gr.Blocks(css="style.css", theme="NoCrypt/miku@1.2.1") as demo:
|
|
| 392 |
)
|
| 393 |
|
| 394 |
if __name__ == "__main__":
|
| 395 |
-
demo.queue(max_size=20).launch(debug=IS_COLAB, share=IS_COLAB)
|
|
|
|
| 30 |
|
| 31 |
MODEL = os.getenv(
|
| 32 |
"MODEL",
|
| 33 |
+
"cagliostrolab/animagine-xl-3.1",
|
| 34 |
)
|
| 35 |
|
| 36 |
torch.backends.cudnn.deterministic = True
|
|
|
|
| 44 |
"madebyollin/sdxl-vae-fp16-fix",
|
| 45 |
torch_dtype=torch.float16,
|
| 46 |
)
|
| 47 |
+
pipeline = StableDiffusionXLPipeline.from_pretrained
|
|
|
|
|
|
|
|
|
|
|
|
|
| 48 |
|
| 49 |
pipe = pipeline(
|
| 50 |
model_name,
|
| 51 |
vae=vae,
|
| 52 |
torch_dtype=torch.float16,
|
| 53 |
custom_pipeline="lpw_stable_diffusion_xl",
|
|
|
|
|
|
|
| 54 |
use_auth_token=HF_TOKEN,
|
| 55 |
)
|
| 56 |
|
|
|
|
| 386 |
)
|
| 387 |
|
| 388 |
if __name__ == "__main__":
|
| 389 |
+
demo.queue(max_size=20).launch(debug=IS_COLAB, share=IS_COLAB)
|