ysharma HF staff commited on
Commit
07a95ca
1 Parent(s): 7c0ff63
Files changed (1) hide show
  1. plms.py +1 -1
plms.py CHANGED
@@ -18,7 +18,7 @@ class PLMSSampler(object):
18
  def register_buffer(self, name, attr):
19
  if type(attr) == torch.Tensor:
20
  if attr.device != torch.device("cuda"):
21
- attr = attr.to(torch.device("cuda"))
22
  setattr(self, name, attr)
23
 
24
  def make_schedule(self, ddim_num_steps, ddim_discretize="uniform", ddim_eta=0., verbose=True):
 
18
  def register_buffer(self, name, attr):
19
  if type(attr) == torch.Tensor:
20
  if attr.device != torch.device("cuda"):
21
+ print("hello") #attr = attr.to(torch.device("cuda"))
22
  setattr(self, name, attr)
23
 
24
  def make_schedule(self, ddim_num_steps, ddim_discretize="uniform", ddim_eta=0., verbose=True):