File size: 1,825 Bytes
273991e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 |
---
base_model:
- THUDM/CogView4-6B
datasets:
- multimodalart/1920-raider-waite-tarot-public-domain
library_name: diffusers
license: other
license_link: https://huggingface.co/THUDM/CogView4-6B/blob/main/LICENSE
widget:
- text: >-
TRTCRD a trtcrd of a knight mounting a running horse wearing an armor and holding a staff, \"knight of wands\"
output:
url: final-5000-0-2-TRTCRD-a-trtcrd-of-a-knig-1741227419.png
- text: >-
TRTCRD a trtcrd of a woman sitting on a throne, wearing a crown and holding a trophee, \"queen of cups\"
output:
url: final-5000-1-2-TRTCRD-a-trtcrd-of-a-woma-1741227417.png
- text: >-
TRTCRD a trtcrd of a person in a red robe holding a scale and giving coins to two kneeling figures, surrounded by six pentacles
output:
url: final-5000-1-2-TRTCRD-a-trtcrd-of-a-pers-1741227455.png
tags:
- text-to-image
- diffusers-training
- diffusers
- template:sd-lora
- cogview4
---
<Gallery />
This is a LoRA fine-tune of the [THUDM/CogView4-6B](https://huggingface.co/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](https://wandb.ai/aryanvs/finetrainers-cogview4).
NOTE: this checkpoint uses sigmas logit_normal weighting. For shifted_sigmas logit_normal weighting, check https://huggingface.co/finetrainers/CogView4-6B-rider-waite-tarot-v0-shifted-sigmas |