Spaces:
Running
on
Zero
Running
on
Zero
Update difpoint/inference.py
Browse files- 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('./
|
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())
|