Create README.md

README.md

@@ -0,0 +1,16 @@
+```model = AutoModelForCausalLM.from_pretrained(
+    model_name,
+    low_cpu_mem_usage=True,
+    return_dict=True,
+    torch_dtype=torch.float16,
+    device_map=device_map,
+)
+tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
+prompt = "menulis esai tentang perekonomian indonesia"
+pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, do_sample=True,
+        top_k=10,
+        num_return_sequences=1,
+        eos_token_id=tokenizer.eos_token_id,
+        max_length=1028, )
+result = pipe(f"<s>[INST] {prompt} [/INST]")
+print(result[0]['generated_text'])```