add default value for bin tokenizer

liamkboyle · May 20, 2024 · cab0ffa · cab0ffa
1 parent 3918d3e
commit cab0ffa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/octo/model/components/tokenizers.py b/octo/model/components/tokenizers.py
@@ -242,7 +242,7 @@ class BinTokenizer(nn.Module):
         high (float): Upper bound for bin range.
     """
 
-    n_bins: int
+    n_bins: int = 256
     bin_type: str = "uniform"
     low: float = 0
     high: float = 1