File size: 223 Bytes
b3d5553 |
1 2 3 4 5 6 7 |
from transformers.models.llama.modeling_flax_llama import (
FlaxLlamaForCausalLM as FlaxCognitivessForCausalLM,
FlaxLlamaModel as FlaxCognitivessModel,
)
# You can add more specific code or changes here if needed.
|