Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
2
3# Chargement du modèle
4tokenizer = AutoTokenizer.from_pretrained("ZigZeug/logllm-llama3-8b-BGL-logs")
5model = AutoModelForCausalLM.from_pretrained("ZigZeug/logllm-llama3-8b-BGL-logs")
6
7# Création du pipeline
8pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
9
10