commit

Former-commit-id: 8af21e8704b199f29ac59d82fe37d872de3c90e0 [formerly 8af21e8704b199f29ac59d82fe37d872de3c90e0 [formerly 8af21e8704b199f29ac59d82fe37d872de3c90e0 [formerly 8af21e8704b199f29ac59d82fe37d872de3c90e0 [formerly 8af21e8704b199f29ac59d82fe37d872de3c90e0 [formerly 1ad46af96222172330ca4a1361e94228cdc385a4]]]]]
Former-commit-id: 6914280c2386f422e333da5824527c519b05d9cc
Former-commit-id: d92a847a71cb6fc030979214cc75245e9d1cb45f

Former-commit-id: 4634f78eea39a6e99f7d62cf24f83ad544283214
Former-commit-id: 72ed2e43abdee7b50825ea7ac4701534f4ffca66
Former-commit-id: 30e20bd953f168881ba2371ee110eb6db5178169
Former-commit-id: df75e1f7a56254eb9f84a028fdf35d49a185e8e8
Former-commit-id: b00c8eaeb479754fcbd396338af147a84bc91b9f

api/routers/sdxl_text_to_image.py

@@ -1,13 +1,27 @@
+<<<<<<< HEAD:api/routers/sdxl_text_to_image.py
+=======
+import sys
+sys.path.append("../scripts")  # Path of the scripts directory
+>>>>>>> 846a4a3 (commit):product_diffusion_api/routers/sdxl_text_to_image.py
 import config
 from fastapi import APIRouter, HTTPException
 from typing import List
 from diffusers import DiffusionPipeline
 import torch
 from functools import lru_cache
+<<<<<<< HEAD:api/routers/sdxl_text_to_image.py
 from scripts.api_utils import accelerator
 from models.sdxl_input import InputFormat
 from async_batcher.batcher import AsyncBatcher
 from scripts.api_utils import pil_to_b64_json, pil_to_s3_json
+=======
+from s3_manager import S3ManagerService
+from PIL import Image
+import io
+
+
+
+>>>>>>> 846a4a3 (commit):product_diffusion_api/routers/sdxl_text_to_image.py
 torch._inductor.config.conv_1x1_as_mm = True
 torch._inductor.config.coordinate_descent_tuning = True
 torch._inductor.config.epilogue_fusion = False
@@ -20,10 +34,38 @@ device = accelerator()
 router = APIRouter()
 
 
+<<<<<<< HEAD:api/routers/sdxl_text_to_image.py
+=======
+
+
+
+def pil_to_b64_json(image):
+    image_id = str(uuid.uuid4())
+    buffered = BytesIO()
+    image.save(buffered, format="PNG")
+    b64_image = base64.b64encode(buffered.getvalue()).decode("utf-8")
+    return {"image_id": image_id, "b64_image": b64_image}
+
+
+def pil_to_s3_json(image: Image.Image,file_name) -> str:
+    image_id = str(uuid.uuid4())
+    s3_uploader = S3ManagerService()
+    image_bytes = io.BytesIO()
+    image.save(image_bytes, format="PNG")
+    image_bytes.seek(0)
+
+    unique_file_name = s3_uploader.generate_unique_file_name(file_name)
+    s3_uploader.upload_file(image_bytes, unique_file_name)
+    signed_url = s3_uploader.generate_signed_url(
+        unique_file_name, exp=43200
+    )  # 12 hours
+    return {"image_id": image_id, "url": signed_url}
+>>>>>>> 846a4a3 (commit):product_diffusion_api/routers/sdxl_text_to_image.py
 
 
 # Load the diffusion pipeline
 @lru_cache(maxsize=1)
+<<<<<<< HEAD:api/routers/sdxl_text_to_image.py
 def load_pipeline(model_name, adapter_name,enable_compile:bool):
     """
     Load the diffusion pipeline with the specified model and adapter names.
@@ -43,6 +85,19 @@ def load_pipeline(model_name, adapter_name,enable_compile:bool):
     if enable_compile is True:
         pipe.unet = torch.compile(pipe.unet, mode="max-autotune")
         pipe.vae.decode = torch.compile(pipe.vae.decode, mode="max-autotune")
+=======
+def load_pipeline(model_name, adapter_name,adapter_name_2):
+    pipe = DiffusionPipeline.from_pretrained(model_name, torch_dtype= torch.bfloat16 ).to(
+        "cuda"
+    )
+    pipe.load_lora_weights(adapter_name)
+    pipe.load_lora_weights(adapter_name_2)
+    pipe.set_adapters([adapter_name, adapter_name_2], adapter_weights=[0.7, 0.5])
+    pipe.fuse_lora()
+    pipe.unload_lora_weights()
+    pipe.unet.to(memory_format=torch.channels_last)
+    pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead")
+>>>>>>> 846a4a3 (commit):product_diffusion_api/routers/sdxl_text_to_image.py
     pipe.fuse_qkv_projections()
     return pipe
 

scripts/output/README.md

@@ -0,0 +1,33 @@
+---
+license: creativeml-openrail-m
+library_name: diffusers
+tags:
+- stable-diffusion-xl
+- stable-diffusion-xl-diffusers
+- text-to-image
+- diffusers
+- diffusers-training
+base_model: stabilityai/stable-diffusion-xl-base-1.0
+inference: true
+---
+
+<!-- This model card has been generated automatically according to the information the training script had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+SDXL Product Images
+
+## Intended uses & limitations
+
+#### How to use
+
+```python
+# TODO: add an example code snippet for running this diffusion pipeline
+```
+
+#### Limitations and bias
+
+[TODO: provide examples of latent issues and potential remediations]
+
+## Training details
+
+[TODO: describe the data used to train the model]

scripts/output/pytorch_lora_weights.safetensors.REMOVED.git-id

@@ -1 +1 @@
-dd6cb78a89b6e91ebaa670d379e4ead3b73ee550
+55f1b147723f81304a3217185314d068b6d49439