VikramSingh178 commited on
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
- # pipe.unet = torch.compile(
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