Zero123plus V1.1

🚀 Zero123plus-v1.1 項目

Zero123plus-v1.1 是基於 diffusers 庫的圖像到圖像生成模型，使用 allenai/objaverse 數據集進行訓練，可用於藝術圖像生成等領域。

🚀 快速開始

推薦使用 diffusers 版本 0.20.2 和 torch 版本 2。

📦 安裝指南

文檔未提及具體安裝步驟，可根據推薦版本使用以下命令安裝：

pip install diffusers==0.20.2 torch==2

💻 使用示例

基礎用法

import torch
import requests
from PIL import Image
from diffusers import DiffusionPipeline, EulerAncestralDiscreteScheduler

# Load the pipeline
pipeline = DiffusionPipeline.from_pretrained(
    "sudo-ai/zero123plus-v1.1", custom_pipeline="sudo-ai/zero123plus-pipeline",
    torch_dtype=torch.float16
)
# Feel free to tune the scheduler
pipeline.scheduler = EulerAncestralDiscreteScheduler.from_config(
    pipeline.scheduler.config, timestep_spacing='trailing'
)
pipeline.to('cuda:0')
# Run the pipeline
cond = Image.open(requests.get("https://d.skis.ltd/nrp/sample-data/lysol.png", stream=True).raw)
result = pipeline(cond).images[0]
result.show()
result.save("output.png")

📄 許可證

本項目使用 openrail 許可證。

📋 信息表格