Hugging Face's logo

Spaces:
VikramSingh178
/
picpilot-server
Runtime error

App Files Community
VikramSingh178 commited on
Commit
6fc3ce6
•
1 Parent(s): 32a04ae

Update SDXL-LoRA inference pipeline and model weights

Browse files

Former-commit-id: e1329aaee0b2059e5d79502309296e7c59f4620a

Files changed (3) hide show
  1. product_diffusion_api/routers/__pycache__/sdxl_text_to_image.cpython-310.pyc +0 -0
  2. product_diffusion_api/routers/sdxl_text_to_image.py +4 -2
  3. scripts/wandb/debug-internal.log +1 -1
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
@@ -52,9 +52,11 @@ class SDXLLoraInference:
52
  guidance_scale: float,
53
  ) -> None:
54
  self.pipe = DiffusionPipeline.from_pretrained(
55
- "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.bfloat16
56
  )
57
- self.pipe.to("cuda")
 
 
58
  self.prompt = prompt
59
  self.negative_prompt = negative_prompt
60
  self.num_images = num_images
 
52
  guidance_scale: float,
53
  ) -> None:
54
  self.pipe = DiffusionPipeline.from_pretrained(
55
+ "stabilityai/stable-diffusion-xl-base-1.0", torch_dtype=torch.float16
56
  )
57
+ self.model_path = "VikramSingh178/sdxl-lora-finetune-product-caption"
58
+ self.pipe.load_lora_weights(self.model_path)
59
+ self.pipe.to('cuda')
60
  self.prompt = prompt
61
  self.negative_prompt = negative_prompt
62
  self.num_images = num_images
scripts/wandb/debug-internal.log CHANGED
@@ -1 +1 @@
1
- run-20240430_104236-lcgqwfyr/logs/debug-internal.log
 
1
+ run-20240501_113730-xllytrin/logs/debug-internal.log