clean up

beast.py

@@ -12,13 +12,6 @@ from huggingface_hub import HfApi
 
 from functools import wraps
 
-# def autocast_float32(fn):
-#     @wraps(fn)
-#     def wrapped(*args, **kwargs):
-#         with torch.cuda.amp.autocast(dtype=torch.float32):
-#             return fn(*args, **kwargs)
-#     return wrapped
-
 def autocast_float32(fn):
     @wraps(fn)
     def wrapped(*args, **kwargs):

beast_params.json

@@ -1,12 +0,0 @@
-{
-  "device": "cuda",
-  "end_cond_order": 0,
-  "enforce_init_pos": false,
-  "gripper_dof": 0,
-  "init_cond_order": 0,
-  "num_basis": 10,
-  "num_dof": 7,
-  "seq_len": 50,
-  "degree_p": 4,
-  "vocab_size": 256
-}

processor_config.json

@@ -2,5 +2,15 @@
   "processor_class": "BeastTokenizer",
   "auto_map":{
     "AutoProcessor": "beast.BeastTokenizer"
-  }
+  },
+  "device": "cuda",
+  "end_cond_order": 0,
+  "enforce_init_pos": false,
+  "gripper_dof": 0,
+  "init_cond_order": 0,
+  "num_basis": 10,
+  "num_dof": 7,
+  "seq_len": 50,
+  "degree_p": 4,
+  "vocab_size": 256
 }