pytorch / serve

Serve, optimize and scale PyTorch models in production
https://pytorch.org/serve/
Apache License 2.0
4.22k stars 863 forks source link

Adding Multi-Image generation usecase app #3356

Open ravi9 opened 2 weeks ago

ravi9 commented 2 weeks ago

Description

This Streamlit app is designed to generate multiple images based on a provided text prompt. It leverages TorchServe for efficient model serving and management, and utilizes Meta-LLaMA-3.2 for prompt generation, and Stable Diffusion with latent-consistency/lcm-sdxl and Torch.compile using OpenVINO backend for image generation.