From a94ec1577a11b56d98f7a703b0d941f17ba1f8af Mon Sep 17 00:00:00 2001 From: Johannes Date: Wed, 9 Oct 2024 18:50:47 +0200 Subject: [PATCH] small fix to disable future warning with torch.load --- enformer_pytorch/modeling_enformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/enformer_pytorch/modeling_enformer.py b/enformer_pytorch/modeling_enformer.py index 9007878..68e655f 100644 --- a/enformer_pytorch/modeling_enformer.py +++ b/enformer_pytorch/modeling_enformer.py @@ -26,7 +26,7 @@ # solution came from @johahi DIR = Path(__file__).parents[0] -TF_GAMMAS = torch.load(str(DIR / "precomputed"/ "tf_gammas.pt")) +TF_GAMMAS = torch.load(str(DIR / "precomputed"/ "tf_gammas.pt"), weights_only=True) # helpers