如何使用 Ovis Image - 文生图快速上手指南

pip install git+https://github.com/DoctorKey/diffusers.git@ovis-image

import torch
from diffusers import OvisImagePipeline

pipe = OvisImagePipeline.from_pretrained(
    "AIDC-AI/Ovis-Image-7B",
    torch_dtype=torch.bfloat16,
)
pipe.to("cuda")

prompt = (
    'A creative 3D artistic render where the text "OVIS-IMAGE" '
    "is written in a bold, expressive handwritten brush style using thick, wet oil paint. "
    "The paint is a mix of vibrant rainbow colors swirling together like toothpaste. "
    "The background is a clean artist's canvas with soft shadows and glossy texture, 4k detail."
)

image = pipe(prompt, negative_prompt="", num_inference_steps=50, true_cfg_scale=5.0).images[0]
image.save("ovis_image.png")

git clone https://github.com/AIDC-AI/Ovis-Image.git
conda create -n ovis-image python=3.10 -y
conda activate ovis-image
cd Ovis-Image
pip install -r requirements.txt
pip install -e .

python ovis_image/test.py \
  --model_path AIDC-AI/Ovis-Image-7B/ovis_image.safetensors \
  --vae_path AIDC-AI/Ovis-Image-7B/ae.safetensors \
  --ovis_path AIDC-AI/Ovis-Image-7B/Ovis2.5-2B \
  --image_size 1024 \
  --denoising_steps 50 \
  --cfg_scale 5.0 \
  --prompt "A creative 3D artistic render where the text \"OVIS-IMAGE\" is written in a bold, expressive handwritten brush style using thick, wet oil paint. The background is a clean artist's canvas with soft shadows, 4k detail."