Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3# Load model dan tokenizer
4model = AutoModelForCausalLM.from_pretrained("67a82a4dd75d090a0e6fb342/TinyLlama-Fantasi-Romansa")
5tokenizer = AutoTokenizer.from_pretrained("67a82a4dd75d090a0e6fb342/TinyLlama-Fantasi-Romansa")
6
7# Contoh prompt
8prompt = "### Instruksi:\nBuatkan pembukaan cerita tentang seorang pangeran yang tersesat di hutan ajaib.\n\n### Respon:\n"
9
10# Generate teks
11inputs = tokenizer(prompt, return_tensors="pt")
12outputs = model.generate(**inputs, max_length=200, temperature=0.7, do_sample=True)
13result = tokenizer.decode(outputs[0], skip_special_tokens=True)
14print(result)