赞
踩
https://huggingface.co/docs/diffusers/tutorials/autopipelinehttps://huggingface.co/docs/diffusers/tutorials/autopipelineAutoPipeline会自动检测要使用的正确流程类,这样可以更轻松地加载与任务相对应的检查点,而无需知道具体的流程类名称。
1.choose an autopipeline for your task
- from diffusers import AutoPipelineForText2Image
- import torch
-
- pipeline = AutoPipelineForText2Image.from_pretrained(
- "runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16, use_safetensors=True
- ).to("cud
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。