VikramxD/PicPilot

Generate Stunning Images and Craft Visual Stories for your Brand

73

stars

398

commits

Python

primary language

Oct 25, 2024

updated

deep-learning
diffusers
docker
generative-ai
litserve
machine-learning
mlops
stable-diffusion
transformers
wandb
webui
Browse cluster: Stable Diffusion UI & Extensions

README

Picpilot Amplify Dark

GitHub Stars License

PicPilot is a scalable solution that leverages state-of-the-art Text to Image Models to extend and enhance images and create product photography in seconds for your brand. Whether you're working with existing product images or need to generate new visuals, PicPilot prepares your visual content to be stunning, professional, and ready for marketing applications.

Features

✅ Flux Inpainting for detailed image editing
✅ SDXL (Stable Diffusion XL) with LoRA for high-quality image generation
✅ SDXL Outpainting for extending images seamlessly
✅ Image to Video Generation using CogvideoX
✅ Batch processing support with configurable batch sizes and timeouts

Why PicPilot?

Creating professional product photography and visual narratives can be time-consuming and expensive. PicPilot aims to revolutionize this process by offering an AI-powered platform where you can enhance existing images, generate new ones, or even convert images to videos, creating stunning visuals for your brand in seconds.

Installation

git clone https://github.com/VikramxD/Picpilot
cd Picpilot

Install Dependencies:

./run.sh

🛳️ Docker

To use PicPilot with Docker, execute the following commands:

docker pull vikram1202/picpilot:latest
docker run --gpus all -p 8000:8000 vikram1202/picpilot:latest

Alternatively, if you prefer to build the Docker image locally:

docker build -t picpilot .
docker run --gpus all -p 8000:8000 picpilot

Usage

Run the Server:

cd api
python picpilot.py

This will start the server on port 8000 with all available API endpoints.

API Endpoints

PicPilot offers the following API endpoints:

EndpointPathPurposeMax Batch SizeBatch Timeout
Flux Inpainting/api/v2/painting/fluxDetailed image editing and inpainting40.1 seconds
SDXL Generation/api/v2/generate/sdxlHigh-quality image generation using SDXL with LoRAConfigured in tti_settingsConfigured in tti_settings
SDXL Outpainting/api/v2/painting/sdxl_outpaintingExtending images seamlessly40.1 seconds
Image to Video/api/v2/image2video/cogvideoxConverting images to videos10.1 seconds

Next Features

  • Support for Image Editing in FLUX Models
  • Support for Custom Flux LORA'S
  • Support for CogvideoX finetuning

Limitations

  • Requires Powerful GPU's to Run for optimal performance Especially the FLUX Models
  • Processing time may vary depending on the complexity of the task and input size
  • Image to video conversion is limited to one image at a time

License

PicPilot is licensed under the MIT license. See LICENSE for more information.

Acknowledgements

This project utilizes several open-source models and libraries. We express our gratitude to the authors and contributors of:

  • Diffusers
  • LitServe
  • Transformers

Contributors

VikramxD

398 commits

VikramxD/PicPilot

Generate Stunning Images and Craft Visual Stories for your Brand

73

stars

398

commits

Python

primary language

Oct 25, 2024

updated

deep-learning
diffusers
docker
generative-ai
litserve
machine-learning
mlops
stable-diffusion
transformers
wandb
webui
Browse cluster: Stable Diffusion UI & Extensions

README

Picpilot Amplify Dark

GitHub Stars License

PicPilot is a scalable solution that leverages state-of-the-art Text to Image Models to extend and enhance images and create product photography in seconds for your brand. Whether you're working with existing product images or need to generate new visuals, PicPilot prepares your visual content to be stunning, professional, and ready for marketing applications.

Features

✅ Flux Inpainting for detailed image editing
✅ SDXL (Stable Diffusion XL) with LoRA for high-quality image generation
✅ SDXL Outpainting for extending images seamlessly
✅ Image to Video Generation using CogvideoX
✅ Batch processing support with configurable batch sizes and timeouts

Why PicPilot?

Creating professional product photography and visual narratives can be time-consuming and expensive. PicPilot aims to revolutionize this process by offering an AI-powered platform where you can enhance existing images, generate new ones, or even convert images to videos, creating stunning visuals for your brand in seconds.

Installation

git clone https://github.com/VikramxD/Picpilot
cd Picpilot

Install Dependencies:

./run.sh

🛳️ Docker

To use PicPilot with Docker, execute the following commands:

docker pull vikram1202/picpilot:latest
docker run --gpus all -p 8000:8000 vikram1202/picpilot:latest

Alternatively, if you prefer to build the Docker image locally:

docker build -t picpilot .
docker run --gpus all -p 8000:8000 picpilot

Usage

Run the Server:

cd api
python picpilot.py

This will start the server on port 8000 with all available API endpoints.

API Endpoints

PicPilot offers the following API endpoints:

EndpointPathPurposeMax Batch SizeBatch Timeout
Flux Inpainting/api/v2/painting/fluxDetailed image editing and inpainting40.1 seconds
SDXL Generation/api/v2/generate/sdxlHigh-quality image generation using SDXL with LoRAConfigured in tti_settingsConfigured in tti_settings
SDXL Outpainting/api/v2/painting/sdxl_outpaintingExtending images seamlessly40.1 seconds
Image to Video/api/v2/image2video/cogvideoxConverting images to videos10.1 seconds

Next Features

  • Support for Image Editing in FLUX Models
  • Support for Custom Flux LORA'S
  • Support for CogvideoX finetuning

Limitations

  • Requires Powerful GPU's to Run for optimal performance Especially the FLUX Models
  • Processing time may vary depending on the complexity of the task and input size
  • Image to video conversion is limited to one image at a time

License

PicPilot is licensed under the MIT license. See LICENSE for more information.

Acknowledgements

This project utilizes several open-source models and libraries. We express our gratitude to the authors and contributors of:

  • Diffusers
  • LitServe
  • Transformers

Contributors

VikramxD

398 commits

Languages

Python

98.2%

Dockerfile

1.3%