Spaces:
Runtime error
Runtime error
Commit
•
6fc3ce6
1
Parent(s):
32a04ae
Update SDXL-LoRA inference pipeline and model weights
Browse filesFormer-commit-id: e1329aaee0b2059e5d79502309296e7c59f4620a
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.
|
56 |
)
|
57 |
-
self.
|
|
|
|
|
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-
|
|
|
1 |
+
run-20240501_113730-xllytrin/logs/debug-internal.log
|