如何解决LMSDiscreteScheduler中.nonzero()返回多元素张量的报错？

这个错误的核心原因是：(self.timesteps == timestep).nonzero()返回的张量包含多个元素（或空张量），而.item()方法仅支持从单元素张量中提取Python标量，因此触发报错。结合你的环境和排查信息，可按以下步骤解决：

一、临时修复代码逻辑 #

直接修改scheduling_lms_discrete.py中的报错行，替换为安全的索引取值方式：

• 若要取第一个匹配的索引：

  idx = (self.timesteps == timestep).nonzero()[0].item()
  

• 若需兼容多匹配场景（或避免空张量报错），可改用torch.where：

  match_indices = torch.where(self.timesteps == timestep)[0]
  idx = match_indices[0].item() if len(match_indices) > 0 else -1
  

• 也可直接转为列表取值：

  idx = (self.timesteps == timestep).nonzero().squeeze().tolist()
  # 若返回是单个值，tolist会自动转成Python标量；若为多个值，需按需处理
  

二、排查版本兼容性问题 #

你的Python和Torch版本（3.8.16 + 1.13.0+cu116）本身与Stable Diffusion兼容，但报错大概率和diffusers库的旧版本bug有关：

1. 升级相关依赖到最新稳定版：

   pip install --upgrade diffusers transformers accelerate
   

2. 验证timestep数据类型：检查self.timesteps和timestep的dtype是否一致（比如一个是int32，一个是int64），若不一致可统一转换：

   timestep = timestep.to(self.timesteps.dtype)
   

三、额外验证步骤 #

• 在循环中打印每次迭代的timestep值和self.timesteps的内容，确认是否存在重复的timestep值，或timestep超出self.timesteps的范围
• 修改代码后，打印idx的值，确保每次迭代都能取到有效的单元素索引

内容的提问来源于stack exchange，提问作者Jan