Views
No views yet
1import torch
2from transformers import pipeline
3
4model_id = "digo-prayudha/Llama-3.2-1B-Indonesian"
5pipe = pipeline(
6 "text-generation",
7 model=model_id,
8 torch_dtype=torch.bfloat16,
9 device_map="auto",
10)
11messages = [
12 {"role": "user", "content": "Tentukan subjek dari kalimat berikut: 'Film tersebut dirilis kemarin'."},
13]
14outputs = pipe(
15 messages,
16 max_new_tokens=256,
17)
18print(outputs[0]["generated_text"][-1])