fix device:

lllyasviel · Mar 4, 2024 · 420013e · 420013e
1 parent d72347e
commit 420013e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ldm_patched/modules/model_sampling.py b/ldm_patched/modules/model_sampling.py
@@ -82,7 +82,7 @@ def timestep(self, sigma):
  w = (low - log_sigma) / (low - high)
  w = w.clamp(0, 1)
  t = (1 - w) * low_idx + w * high_idx
- return t.view(sigma.shape)
+ return t.view(sigma.shape).to(sigma.device)
 
  def sigma(self, timestep):
  t = torch.clamp(timestep.float().to(self.log_sigmas.device), min=0, max=(len(self.sigmas) - 1))