Skip to content

Commit

Permalink
Default device for sigma tensor to CPU
Browse files Browse the repository at this point in the history
* Consistent with implementations in k-diffusion.
* Makes this compatible with AUTOMATIC1111#15823
  • Loading branch information
LoganBooker committed May 17, 2024
1 parent d6b4444 commit 1d74482
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion modules/sd_schedulers.py
Original file line number Diff line number Diff line change
Expand Up @@ -33,7 +33,7 @@ def sgm_uniform(n, sigma_min, sigma_max, inner_model, device):
sigs += [0.0]
return torch.FloatTensor(sigs).to(device)

def get_align_your_steps_sigmas(n, sigma_min, sigma_max, device):
def get_align_your_steps_sigmas(n, sigma_min, sigma_max, device='cpu'):
# https://research.nvidia.com/labs/toronto-ai/AlignYourSteps/howto.html
def loglinear_interp(t_steps, num_steps):
"""
Expand Down

0 comments on commit 1d74482

Please sign in to comment.