JaehaerysNL/StableDiffusion
1
0
Fork 0
mirror of https://github.com/Stability-AI/stablediffusion.git synced 2024-12-22 23:55:00 +00:00
Code Issues Projects Releases Packages Wiki Activity

Separate mps and other processes

Browse source
This commit is contained in:
Ftps 2023-02-09 16:38:18 +09:00
parent 19697bbf07
commit a101942e8a
1 changed files with 4 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
ldm/models/diffusion/ddim.py
View file

@ -18,7 +18,10 @@ class DDIMSampler(object):
def register_buffer(self, name, attr):
if type(attr) == torch.Tensor:
if attr.device != self.device:
attr = attr.to(self.device, torch.float32)
if str(self.device) == 'mps':
attr = attr.to(self.device, torch.float32)
else:
attr = attr.to(self.device)
setattr(self, name, attr)
def make_schedule(self, ddim_num_steps, ddim_discretize="uniform", ddim_eta=0., verbose=True):