SigmaJourney - Pixart E + MJv6

This is a finetuned version of Pixart Sigma (E).

Improvement in aesthetic

Enhanced MidJourney v6 Style

Usage:

SD NEXT: https://github.com/vladmandic/automatic
ComfyUI with ExtraModels Node: https://github.com/city96/ComfyUI_ExtraModels
Diffusers pipeline

import torch

from diffusers import DiffusionPipeline, EulerAncestralDiscreteScheduler

from diffusers.models import PixArtTransformer2DModel





model_id = "TensorFamily/SigmaJourney"

negative_prompt = "malformed, disgusting, overexposed, washed-out"



pipeline = DiffusionPipeline.from_pretrained(model_id, torch_dtype=torch.float16)

pipeline.scheduler = EulerAncestralDiscreteScheduler.from_config(pipeline.scheduler.config)

pipeline.to('cuda' if torch.cuda.is_available() else 'cpu')



prompt = "On the left, there is a red cube. On the right, there is a blue sphere. On top of the red cube is a dog. On top of the blue sphere is a cat"

image = pipeline(

    prompt=prompt,

    negative_prompt='blurry, cropped, ugly',

    num_inference_steps=30,

    generator=torch.Generator(device='cuda' if torch.cuda.is_available() else 'mps' if torch.backends.mps.is_available() else 'cpu').manual_seed(1641421826),

    width=1024,

    height=1024,

    guidance_scale=5.5,

).images[0]

image.save("output.jpg")