Open shmishra99 opened 9 months ago
import os os.environ["KERAS_BACKEND"] = "tensorflow" gemma_lm = keras_nlp.models.GemmaCausalLM.from_preset("gemma_instruct_7b_en") gotA = gemma_lm.generate("Keras is a", max_length=30) print(gotA)