You need to enable JavaScript to run this app.
优惠活动
大模型
产品
解决方案
定价
更多
文档控制台
免费开始使用

运行Z-Image-Turbo示例时Huggingface缓存下载无报错停滞求助

Z-Image-Turbo模型下载停滞问题排查

运行Z-Image-Turbo模型时,执行以下示例代码:

import os
os.environ["HF_HOME"] = "E:/MyCustomHFCache"
os.environ["TRANSFORMERS_CACHE"] = "E:/MyTransformersCache"

import torch
from diffusers import ZImagePipeline

# 1. 加载pipeline
# 在支持的GPU上使用bfloat16获得最佳性能
pipe = ZImagePipeline.from_pretrained(
    "Tongyi-MAI/Z-Image-Turbo",
    torch_dtype=torch.bfloat16,
    low_cpu_mem_usage=False,
)
pipe.to("cuda")

prompt = "身着红色汉服的年轻中国女子,衣身绣工繁复。妆容精致,额间饰有红色花钿。梳着精致高发髻,头戴金凤头饰,搭配红花与珠饰。手持绘有仕女、树木、飞鸟的团扇。左手掌心上方悬浮着霓虹闪电灯(⚡️),发出明亮黄光。背景为柔和光线的户外夜景,远处可见西安大雁塔的剪影,还有模糊的彩色灯光。"

# 2. 生成图像
image = pipe(
    prompt=prompt,
    height=1024,
    width=1024,
    num_inference_steps=9,  # 实际对应8次DiT前向传播
    guidance_scale=0.0,     # Turbo模型需将guidance设置为0
    generator=torch.Generator("cuda").manual_seed(42),
).images[0]

image.save("example.png")

在下载约30GB内容后,进度突然停滞,磁盘剩余空间充足,控制台无任何错误提示,也没有网络流量传入。从下图可见多个文件的下载进度状态:

多个文件下载进度条

可能的原因及解决方向:

  • 网络连接中断:Hugging Face Hub的节点可能临时失联,且下载器未自动重试。可手动中断程序后重新运行,或添加环境变量HF_HUB_MAX_RETRIES=5增加重试次数。
  • 断点续传异常:部分大权重文件的临时下载文件(带.part后缀)损坏或卡住。前往设置的缓存目录E:/MyCustomHFCacheE:/MyTransformersCache,删除未完成的临时文件后重新下载。
  • 缓存目录权限不足:即使磁盘空间充足,若当前用户对缓存目录无写入权限,下载会静默终止。检查E盘对应目录的权限,确保用户拥有完全读写权限。
  • 仓库文件同步延迟:Hugging Face仓库内的部分文件存在镜像同步问题,导致无法获取。可设置环境变量切换到国内镜像源。
  • 下载超时设置过短:默认下载超时时间不足以支撑大文件下载,触发超时但无提示。可添加环境变量HF_HUB_DOWNLOAD_TIMEOUT=300(单位:秒)延长超时时间。

内容的提问来源于stack exchange,提问作者Recently_Created_User

火山引擎 最新活动