Update custom.py

david-thrower · web-flow · commit 98645bd7de6b · 2024-06-13T15:03:47.000-04:00
1/2 move imports from keras_nlp.models import GPT2Tokenizer, GPT2Preprocessor, GPT2Backbone
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 
