Modelpublic

finetrainers/CogView4-6B-rider-waite-tarot-v0

sourceHugging Faceotherupdated 2y agoView on Hugging Face

1likes13downloads

Model Card

This is a LoRA fine-tune of the THUDM/CogView4-6B model.

Code: https://github.com/a-r-r-o-w/finetrainers

Inference code:

python

import torch
from diffusers import CogView4Pipeline
from diffusers.utils import export_to_video

pipe = CogView4Pipeline.from_pretrained(
    "THUDM/CogView4-6B", torch_dtype=torch.bfloat16
).to("cuda")
pipe.load_lora_weights("finetrainers/CogView4-6B-rider-waite-tarot-v0", adapter_name="cogview4-lora")
pipe.set_adapters(["cogview4-lora"], [0.9])

image = pipe("<my-awesome-prompt>").images[0]
image.save("output.png")

Training logs are available on WandB here.

NOTE: this checkpoint uses sigmas logitnormal weighting. For shiftedsigmas logit_normal weighting, check https://huggingface.co/finetrainers/CogView4-6B-rider-waite-tarot-v0-shifted-sigmas