msArray's picture
Update README.md
930ba2e verified
|
Raw
History Blame Contribute Delete
1.23 kB
metadata
language:
  - en
tags:
  - text-to-image
  - stable-diffusion
  - safetensors
  - stable-diffusion-xl
license: openrail++
base_model:
  - cagliostrolab/animagine-xl-4.0
base_model_relation: quantized

Animagine XL 4.0 8bit

Example Code

import torch
from diffusers import StableDiffusionXLPipeline
import os
import secrets

pipe = StableDiffusionXLPipeline.from_pretrained(
    "msArray/animagine-xl-4.0-8bit",
    torch_dtype=torch.float16,
    use_safetensors=True,
    device_map="cuda"
)

prompt = "1girl, toki \(blue archive\), blue archive, safe, masterpiece, high score, great score, absurdres"
negative_prompt = "lowres, bad anatomy, bad hands, text, error, missing finger, extra digits, fewer digits, cropped, worst quality, low quality, low score, bad score, average score, signature, watermark, username, blurry"

image = pipe(
    prompt,
    negative_prompt=negative_prompt,
    width=832,
    height=1216,
    guidance_scale=5,
    num_inference_steps=28
).images[0]

if not os.path.exists("./output"):
    os.makedirs("./output")
    
image.save(f"./output/{secrets.token_hex(10)}.png")

image