Downloads · 30 days
7
2% of all-time downloads
concept-caption-3m-sd-lora-zh
svjack/concept-caption-3m-sd-lora-zh
concept-caption-3m-sd-lora-zh is a machine learning model from svjack. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for diffusers.
All-time downloads
408
Public
Repo size
3.3 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.bin3.3 MB · 100%
At a glance
- Library
- diffusers
- Access
- Public
- Created
- Mar 20, 2023
- Updated
- Mar 20, 2023
- SHA
- 12bb7228
Model card
From the Hugging Face model README
Model Card for svjack/concept-caption-3m-sd-lora-zh
Installation
pip install -U diffusers
pip install transformers
Usage
from diffusers import StableDiffusionPipeline
import torch
pipe = StableDiffusionPipeline.from_pretrained("IDEA-CCNL/Taiyi-Stable-Diffusion-1B-Chinese-v0.1", torch_dtype=torch.float16)
model_path = "svjack/concept-caption-3m-sd-lora-zh"
pipe.unet.load_attn_procs(model_path)
pipe.to("cuda")
pipe.safety_checker = lambda images, clip_input: (images, False)
print("have_load")
prompt = "一只快乐的狗。"
image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5).images[0]
image
- Library
- diffusers
- Languages
- zh
- Created
- Mar 20, 2023
- Updated
- Mar 20, 2023