fix

atomicarchitects · Aug 15, 2024 · 6d4c641 · 6d4c641
1 parent b9a28ff
commit 6d4c641
Showing 1 changed file with 0 additions and 4 deletions.
diff --git a/model.py b/model.py
@@ -1,9 +1,5 @@
 import torch
-torch.jit.script = lambda x: x
 import torch.nn as nn
-import e3nn
-e3nn.set_optimization_defaults(jit_script_fx=False)
-
 from e3nn import o3
 
 from torch_geometric.data import Data