wi-lab
/

lwm

Wireless-Communication

Zero-Shot Classification

Feature-Extraction

Inference Endpoints

Model card Files Files and versions Community

wi-lab commited on Sep 26

Commit

9a3c9ce

•

1 Parent(s): 08a8088

Update lwm_model.py

Files changed (1) hide show

lwm_model.py +1 -1

lwm_model.py CHANGED Viewed

@@ -116,7 +116,7 @@ class lwm(torch.nn.Module):
         model = cls().to(device)
         ckpt_path = ckpt_name
-        model.load_state_dict(torch.load(ckpt_path, map_location=device), weights_only=True)
         print(f"Model loaded successfully from {ckpt_path} to {device}")
         return model

         model = cls().to(device)
         ckpt_path = ckpt_name
+        model.load_state_dict(torch.load(ckpt_path, map_location=device))
         print(f"Model loaded successfully from {ckpt_path} to {device}")
         return model