diff --git a/nets/equiformer_v2/equiformer_v2_oc20.py b/nets/equiformer_v2/equiformer_v2_oc20.py
index 20cda55..126c989 100644
--- a/nets/equiformer_v2/equiformer_v2_oc20.py
+++ b/nets/equiformer_v2/equiformer_v2_oc20.py
@@ -390,7 +390,6 @@ def forward(self, data):
         ###############################################################
 
         # Init per node representations using an atomic number based embedding
-        offset = 0
         x = SO3_Embedding(
             num_atoms,
             self.lmax_list,
@@ -530,4 +529,4 @@ def no_weight_decay(self):
                     global_parameter_name = module_name + '.' + parameter_name
                     assert global_parameter_name in named_parameters_list
                     no_wd_list.append(global_parameter_name)
-        return set(no_wd_list)
\ No newline at end of file
+        return set(no_wd_list)