Koios-API/KoiosAPI-codegemma-7b-it

1import tensorflow as tf2from transformers import TFAutoModelForCausalLM3 4# Optionally, test loading a model to ensure all components are working5model = TFAutoModelForCausalLM.from_pretrained("gpt2")  # Just an example, use a model appropriate for TFAutoModelForCausalLM6print("Model loaded successfully!")7 8