artisokka
/

tinyllama-dialogsum-finetuned

dialogue summarization

text2text-generation

Model card Files Files and versions Community

artisokka commited on Mar 13

Commit

a7a818b

·

verified ·

1 Parent(s): 2388c39

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -39,7 +39,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 from peft import PeftModel
 model_name = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
-peft_model_id = "YOUR_HUGGINGFACE_USERNAME/YOUR_REPO_NAME" #replace with your repo name
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)

 from peft import PeftModel
 model_name = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
+peft_model_id = "artisokka/tinyllama-dialogsum-finetuned"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)