LinFusion: 1 GPU, 1 Minute, 16K Image
Paper • 2409.02097 • Published • 34
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("Yuanshi/LinFusion-1-5", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]This model has been pushed to the Hub using the Diffusers library:
- Basic Usage of LinFusion:
from diffusers import AutoPipelineForText2Image
import torch
from src.linfusion import LinFusion
from src.tools import seed_everything
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)
seed_everything(123)
image = pipeline(
"An astronaut floating in space. Beautiful view of the stars and the universe in the background."
).images[0]
- Downloads last month
- 27