Hugging Face's logo

Spaces:
TripletMix
/
tripletmix-demo
Sleeping

App Files Community
winfred2027 commited on
Commit
9ce8dc1
·
verified ·
1 Parent(s): 428cfb2

Update openshape/__init__.py

Browse files
Files changed (1) hide show
  1. openshape/__init__.py +1 -1
openshape/__init__.py CHANGED
@@ -37,7 +37,7 @@ def S14(s):
37
  PointPatchTransformer(256, 6, 4, 1024, 96, 64, 0.4, 256, 6),
38
  nn.Linear(256, 1280)
39
  )
40
- model.load_state_dict(module(['state_dict'], 'module'))
41
  return model
42
 
43
  def G14_M(s):
 
37
  PointPatchTransformer(256, 6, 4, 1024, 96, 64, 0.4, 256, 6),
38
  nn.Linear(256, 1280)
39
  )
40
+ model.load_state_dict(module(s['state_dict'], 'module'))
41
  return model
42
 
43
  def G14_M(s):