Update skyreels_v2_infer/modules/clip.py

skyreels_v2_infer/modules/clip.py

@@ -497,7 +497,7 @@ class CLIPModel(ModelMixin):
         )
         self.model = self.model.eval().requires_grad_(False)
         logging.info(f"loading {checkpoint_path}")
-        self.model.load_state_dict(torch.load(checkpoint_path, map_location="cpu"))
+        self.model.load_state_dict(torch.load(checkpoint_path, map_location="cpu", weights_only=True))
 
         # init tokenizer
         self.tokenizer = HuggingfaceTokenizer(