Spaces:
Running
Running
Commit
•
27d6043
1
Parent(s):
3e01790
chore: Update SDXL-LoRA inference pipeline and model weights
Browse files
.gitignore
CHANGED
@@ -1,5 +1,5 @@
|
|
1 |
.venv
|
2 |
-
scripts/wandb
|
3 |
variables.tf
|
4 |
.terraform
|
5 |
|
|
|
1 |
.venv
|
2 |
+
/scripts/wandb
|
3 |
variables.tf
|
4 |
.terraform
|
5 |
|
product_diffusion_api/routers/__pycache__/sdxl_text_to_image.cpython-310.pyc
CHANGED
Binary files a/product_diffusion_api/routers/__pycache__/sdxl_text_to_image.cpython-310.pyc and b/product_diffusion_api/routers/__pycache__/sdxl_text_to_image.cpython-310.pyc differ
|
|
product_diffusion_api/routers/sdxl_text_to_image.py
CHANGED
@@ -37,13 +37,13 @@ def load_pipeline(model_name, adapter_name):
|
|
37 |
"cuda"
|
38 |
)
|
39 |
pipe.load_lora_weights(adapter_name)
|
|
|
|
|
|
|
|
|
40 |
pipe.unet.to(memory_format=torch.channels_last)
|
41 |
pipe.vae.to(memory_format=torch.channels_last)
|
42 |
-
|
43 |
-
# pipe.unet,
|
44 |
-
# mode = 'max-autotime'
|
45 |
-
# )
|
46 |
-
|
47 |
pipe.fuse_qkv_projections()
|
48 |
|
49 |
return pipe
|
|
|
37 |
"cuda"
|
38 |
)
|
39 |
pipe.load_lora_weights(adapter_name)
|
40 |
+
pipe.unet = torch.compile(
|
41 |
+
pipe.unet,
|
42 |
+
mode = 'max-autotune'
|
43 |
+
)
|
44 |
pipe.unet.to(memory_format=torch.channels_last)
|
45 |
pipe.vae.to(memory_format=torch.channels_last)
|
46 |
+
|
|
|
|
|
|
|
|
|
47 |
pipe.fuse_qkv_projections()
|
48 |
|
49 |
return pipe
|