【求助帖】AutoDL原生stable-diffusion-3.5-medium模型无卡模式启动和使用 #592
Labels
No Label
bug
duplicate
enhancement
help wanted
invalid
question
wontfix
No Milestone
No project
No Assignees
2 Participants
Notifications
Due Date
No due date set.
Dependencies
No dependencies set.
Reference: HswOAuth/llm_course#592
Loading…
Reference in New Issue
Block a user
No description provided.
Delete Branch "%!s()"
Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?
使用modelscope下载stable-diffusion-3.5-medium模型如下

调用代码如下:
import torch
from diffusers import StableDiffusion3Pipeline
#pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-medium", torch_dtype=torch.bfloat16)
pipe = StableDiffusion3Pipeline.from_pretrained("/root/autodl-tmp/Diffusion-Model", torch_dtype=torch.bfloat16)
pipe = pipe.to("cuda")
image = pipe(

"A capybara holding a sign that reads Hello World",
num_inference_steps=40,
guidance_scale=4.5,
).images[0]
image.save("capybara.png")
调用失败如图,是否因为无卡模式原因?
可能是由于内存问题,可以减少prompt数量为1个再试试,不行的话就只能用显存跑了