YinuoGuo27 commited on
Commit
cbffd7f
·
verified ·
1 Parent(s): 19994bd

Update difpoint/inference.py

Browse files
Files changed (1) hide show
  1. difpoint/inference.py +1 -1
difpoint/inference.py CHANGED
@@ -162,7 +162,7 @@ class Inferencer(object):
162
  self.device = 'cuda'
163
  from difpoint.model import get_model
164
  self.point_diffusion = get_model()
165
- ckpt = torch.load('./download_repo/ckpts/KDTalker.pth')
166
 
167
  self.point_diffusion.load_state_dict(ckpt['model'])
168
  print('model', self.point_diffusion.children())
 
162
  self.device = 'cuda'
163
  from difpoint.model import get_model
164
  self.point_diffusion = get_model()
165
+ ckpt = torch.load('./downloaded_repo/ckpts/KDTalker.pth')
166
 
167
  self.point_diffusion.load_state_dict(ckpt['model'])
168
  print('model', self.point_diffusion.children())