Update README.md

README.md

@@ -1,5 +1,5 @@
 ---
-library_name: peft
+library_name: transformers
 license: cc-by-nc-4.0
 language:
 - en
@@ -12,6 +12,7 @@ tags:
 - uncensored
 - instruct
 - alpaca
+pipeline_tag: text-generation
 ---
 # DukunLM - Indonesian Language Model 🧙‍♂️
 
@@ -61,11 +62,7 @@ model = AutoPeftModelForCausalLM.from_pretrained(
         bnb_4bit_quant_type="nf4",
     )
 )
-tokenizer = AutoTokenizer.from_pretrained(
-    "azale-ai/DukunLM-Uncensored-7B",
-    use_fast=False,
-    padding_side="right"
-)
+tokenizer = AutoTokenizer.from_pretrained("azale-ai/DukunLM-Uncensored-7B")
 streamer = TextStreamer(tokenizer)
 
 instruction_prompt = "Jelaskan mengapa air penting bagi kehidupan manusia."
@@ -97,9 +94,8 @@ _ = model.generate(
     streamer=streamer,
     pad_token_id=tokenizer.pad_token_id,
     eos_token_id=tokenizer.eos_token_id,
-    max_length=2048, use_cache=True,
-    temperature=0.7, do_sample=True,
-    top_k=4, top_p=0.95
+    max_length=2048, temperature=0.7,
+    do_sample=True, top_k=4, top_p=0.95
 )
 ```
 
@@ -124,11 +120,7 @@ model = AutoPeftModelForCausalLM.from_pretrained(
         bnb_4bit_quant_type="nf4",
     )
 )
-tokenizer = AutoTokenizer.from_pretrained(
-    "azale-ai/DukunLM-Uncensored-7B",
-    use_fast=False,
-    padding_side="right"
-)
+tokenizer = AutoTokenizer.from_pretrained("azale-ai/DukunLM-Uncensored-7B")
 
 instruction_prompt = "Bangun dialog chatbot untuk layanan pelanggan yang ingin membantu pelanggan memesan produk tertentu."
 input_prompt = "Produk: Sepatu Nike Air Max"
@@ -158,9 +150,8 @@ _ = model.generate(
     inputs=inputs.input_ids,
     pad_token_id=tokenizer.pad_token_id,
     eos_token_id=tokenizer.eos_token_id,
-    max_length=2048, use_cache=True,
-    temperature=0.7, do_sample=True,
-    top_k=4, top_p=0.95
+    max_length=2048, temperature=0.7,
+    do_sample=True, top_k=4, top_p=0.95
 )
 print(tokenizer.decode(outputs[0], skip_special_tokens=True))
 ```