Generalis_V1.1 / README.md
vluz's picture
Update README.md
1fc910e
|
raw
history blame
1.35 kB
metadata
license: cc0-1.0
language:
  - en

Generalis V1.1


Second attempt at merging several models v1.5 into one general purpose model.

Focus has been put into simple prompts, good one-off generation, slightly muted colours, low memory usage, small model size.

It is intended as easy model for use in larger projects where image generation is needed.

Published under CC0


Use example:

import torch    # Tested with 2.0.1+cu118
from diffusers import StableDiffusionPipeline    # <3


# Model location in HF 
model = "https://huggingface.co/vluz/Generalis_V1.1/blob/main/Generalis_v1-1.safetensors"

# Create pipe
pipe = StableDiffusionPipeline.from_ckpt(model, 
    torch_dtype=torch.float16, 
    safety_checker=None,
    feature_extractor=None,
    requires_safety_checker=False,)

# Cleanup
del pipe.vae.encoder
torch.cuda.empty_cache()

# Send to GPU
pipe = pipe.to("cuda")

# Optimize for low vram use and clear cache again
pipe.enable_vae_tiling()
pipe.enable_attention_slicing("max")
pipe.enable_xformers_memory_efficient_attention(attention_op=None)
pipe.unet.to(memory_format=torch.channels_last)
pipe.enable_sequential_cpu_offload()
torch.cuda.empty_cache()

# Set a prompt
prompt = "a cat"

# Generate image based on prompt
image = pipe(prompt).images[0]

# Save result image to disk
image.save("cat.png")