Update test_llm_serialization.py

david-thrower · web-flow · commit de444e4fdc2c · 2025-10-18T05:06:35.000-04:00
More debugging Keras ...
diff --git a/test_llm_serialization.py b/test_llm_serialization.py
@@ -1,7 +1,15 @@
 
 import tensorflow as tf
 from transformers import AutoTokenizer
-from cerebrosllmutils.llm_utils import CerebrosNotGPT  # Required for deserialization of custom model
+from cerebrosllmutils.llm_utils import (
+            RotaryEmbedding,
+            split_alternate,
+            rotate_half,
+            apply_rotary_pos_emb,
+            InterleavedRoPE,
+            Perplexity,
+            CerebrosNotGPTConfig,
+            CerebrosNotGPT)
 
 def test_serialization(tokenizer_path: str, model_path: str):
     try:
