LinFusion-1-5 / README.md
lichorosario's picture
Update README.md
647ed1b verified
|
raw
history blame
779 Bytes
metadata
library_name: diffusers
tags:
  - pytorch_model_hub_mixin
  - model_hub_mixin
pipeline_tag: text-to-image

This model has been pushed to the Hub using ****:

  • Repo: [More Information Needed]
  • Docs: [More Information Needed]
  • 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]