Views
No views yet
"normal" → Expected system behavior."anomalous" → Suspicious or error-prone system behavior.1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3model_id = "ZigZeug/LogLaye-LLaMA3.2B-QLoRA-finetuned-HDFS-logs"
4
5model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", torch_dtype="auto")
6tokenizer = AutoTokenizer.from_pretrained(model_id)
7
8prompt = [
9 {"role": "system", "content": "You are an expert in HDFS log analysis. Classify if the following log is normal or anomalous."},
10 {"role": "user", "content": "Log: PacketResponder 2 for block blk_-3552845605773916309 terminating"}
11]
12
13chat = tokenizer.apply_chat_template(prompt, tokenize=False, add_generation_prompt=True)
14inputs = tokenizer(chat, return_tensors="pt").to("cuda")
15outputs = model.generate(**inputs, max_new_tokens=20)
16
17print(tokenizer.decode(outputs[0], skip_special_tokens=True))