Update infer/worldmodel.py

infer/worldmodel.py

@@ -31,7 +31,7 @@ class Worldinfer():
         else:
             assert False, "currently rwkv7 strategy must be: cuda/cpu fp16/fp32/bf16"
         
-        self.model_weight = torch.load(model_path + '.pth', map_location=DEVICE)
+        self.model_weight = torch.load(model_path, map_location=DEVICE)
         modality_dict = {}
         for key, value in self.model_weight.items():
             if 'emb.weight' in key: