README.md

LinFusion

LinFusion: 1 GPU, 1 Minute, 16K Image
Songhua Liu, Weuhao Yu, Zhenxiong Tan, and Xinchao Wang
Learning and Vision Lab, National University of Singapore

Supported Models

1. Yuanshi/LinFusion-1-5: For Stable Diffusion 1.5 and its variants.

Quick Start

• If you have not, install PyTorch and diffusers.

• Clone this repo to your project directory:

  git clone https://github.com/Huage001/LinFusion.git

• You only need two lines!

  from diffusers import AutoPipelineForText2Image
  import torch
  
  + from src.linfusion import LinFusion
  
  sd_repo = "Lykon/dreamshaper-8"
  
  pipeline = AutoPipelineForText2Image.from_pretrained(
      sd_repo, torch_dtype=torch.float16, variant="fp16"
  ).to(torch.device("cuda"))
  
  + linfusion = LinFusion.construct_for(pipeline)
  
  image = pipeline(
      "An astronaut floating in space. Beautiful view of the stars and the universe in the background.",
      generator=torch.manual_seed(123)
  ).images[0]

  LinFusion.construct_for(pipeline) will return a LinFusion model that matches the pipeline's structure. And this LinFusion model will automatically mount to the pipeline's forward function.

• examples/basic_usage.ipynb shows a basic text-to-image example.

ToDo

• Stable Diffusion 1.5 support.
• Stable Diffusion 2.1 support.
• Stable Diffusion XL support.
• Release training code for LinFusion.
• Release evaluation code for LinFusion.

Citation

If you finds this repo is helpful, please consider cite:

@article{liu2024linfusion,
  title     = {LinFusion: 1 GPU, 1 Minute, 16K Image},
  author    = {Liu, Songhua and Yu, Weihao and Tan, Zhenxiong and Wang, Xinchao},
  year      = {2024},
  eprint    = {2409.02097},
  archivePrefix={arXiv},
  primaryClass={cs.CV}
}