Update skyreels_v2_infer/modules/t5.py

skyreels_v2_infer/modules/t5.py

@@ -434,7 +434,7 @@ class T5EncoderModel(ModelMixin):
         # init model
         model = umt5_xxl(encoder_only=True, return_tokenizer=False)
         logging.info(f"loading {checkpoint_path}")
-        model.load_state_dict(torch.load(checkpoint_path, map_location="cpu"))
+        model.load_state_dict(torch.load(checkpoint_path, map_location="cpu", weights_only=True))
         self.model = model
         if shard_fn is not None:
             self.model = shard_fn(self.model, sync_module_states=False)