markod0925/yarn_flux

Model Card for Model ID

Make everything as Yarn Art Syle.

Uses

Make everything as Yarn Art Syle.

How to Get Started with the Model

!pip install -U diffusers bitsandbytes

from diffusers import AutoPipelineForText2Image, BitsAndBytesConfig
import torch
repo_id = "markod0925/yarn_flux"

bnb_4bit_compute_dtype = torch.float16
nf4_config = BitsAndBytesConfig(
    load_in_4bit=True,
    bnb_4bit_quant_type="nf4",
    bnb_4bit_compute_dtype=bnb_4bit_compute_dtype,
)

pipeline = AutoPipelineForText2Image.from_pretrained(repo_id,
                                                     torch_dtype=bnb_4bit_compute_dtype
)
pipeline.enable_model_cpu_offload()

image = pipeline(
    "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k", num_inference_steps=28, guidance_scale=3.5, height=768
).images[0]
image

image = pipeline(
    "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k, yarn art style", num_inference_steps=28, guidance_scale=3.5, height=768
).images[0]
image