Spaces:
Running
Running
| from typing import Any | |
| from app_settings import Settings | |
| from models.interface_types import InterfaceType | |
| from backend.lcm_text_to_image import LCMTextToImage | |
| from time import perf_counter | |
| from backend.image_saver import ImageSaver | |
| from pprint import pprint | |
| class Context: | |
| def __init__( | |
| self, | |
| interface_type: InterfaceType, | |
| device="cpu", | |
| ): | |
| self.interface_type = interface_type | |
| self.lcm_text_to_image = LCMTextToImage(device) | |
| def generate_text_to_image( | |
| self, | |
| settings: Settings, | |
| reshape: bool = False, | |
| device: str = "cpu", | |
| ) -> Any: | |
| tick = perf_counter() | |
| from state import get_settings | |
| get_settings().save() | |
| pprint(settings.lcm_diffusion_setting.model_dump()) | |
| if not settings.lcm_diffusion_setting.lcm_lora: | |
| return None | |
| self.lcm_text_to_image.init( | |
| device, | |
| settings.lcm_diffusion_setting, | |
| ) | |
| images = self.lcm_text_to_image.generate( | |
| settings.lcm_diffusion_setting, | |
| reshape, | |
| ) | |
| elapsed = perf_counter() - tick | |
| # ImageSaver.save_images( | |
| # settings.results_path, | |
| # images=images, | |
| # lcm_diffusion_setting=settings.lcm_diffusion_setting, | |
| # ) | |
| print(f"Latency : {elapsed:.2f} seconds") | |
| return images | |