Neithanvarvas commited on
Commit
12f89f8
·
verified ·
1 Parent(s): e976cb5

[FEAT] Add CBT and CBTConfig class aliases

Browse files
Files changed (1) hide show
  1. model.py +6 -0
model.py CHANGED
@@ -419,3 +419,9 @@ class ScaledGPT(nn.Module):
419
  break
420
 
421
  return idx
 
 
 
 
 
 
 
419
  break
420
 
421
  return idx
422
+
423
+
424
+ # Aliases for Character Based Transformer (CBT)
425
+ CBT = ScaledGPT
426
+ CBTConfig = ScaledGPTConfig
427
+