Update custom.py

1/2 move imports from keras_nlp.models import GPT2Tokenizer, GPT2Preprocessor, GPT2Backbone
david-thrower · Jun 13, 2024 · 98645bd · 98645bd
1 parent 5e583c7
commit 98645bd
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/cerebros/custom/custom.py b/cerebros/custom/custom.py
@@ -1,5 +1,6 @@
 
 import tensorflow as tf
+from keras_nlp.models import GPT2Tokenizer, GPT2Preprocessor, GPT2Backbone
 
 # Custom keras layer, analogue of Dense that does 
 # ternary mathematical operations