ocpub/oc9-diffusers repository overview

docker build -t oc9-diffusers:0.33.0 .

# 验证 diffusers 版本
docker run --rm --gpus all oc9-diffusers:0.33.0 python3 -c "import diffusers; print(diffusers.__version__)"

# 验证 CUDA 可用性
docker run --rm --gpus all oc9-diffusers:0.33.0 python3 -c "import torch; print(torch.cuda.is_available())"

# 快速推理示例（需要联网下载模型）
docker run --rm --gpus all oc9-diffusers:0.33.0 python3 -c "
from diffusers import DiffusionPipeline
import torch
pipe = DiffusionPipeline.from_pretrained('runwayml/stable-diffusion-v1-5', torch_dtype=torch.float16)
pipe = pipe.to('cuda')
print('Pipeline loaded on GPU')
"

# 执行测试脚本（需要 GPU 环境）
bash test.sh oc9-diffusers:0.33.0