File size: 203 Bytes
bcf8bb7 |
1 2 3 4 5 6 7 8 |
# POC Model
from transformers.models.llama import LlamaForCausalLM
class POCModel(LlamaForCausalLM):
def __init__(self, config):
super().__init__(config)
print("[POC] Model loaded")
|