Update README.md

README.md

@@ -37,7 +37,7 @@ import transformers
 from transformers import PeftModel
 
 model_name = "google/flan-t5-large"
-peft_model_id = "ckpts_databricks_large"
+peft_model_id = "Leadmatic/tinyChat"
 tokenizer = transformers.AutoTokenizer.from_pretrained(model_name)
 base_model = transformers.AutoModelForSeq2SeqLM.from_pretrained(model_name)
 peft_model = PeftModel.from_pretrained(base_model, peft_model_id)
@@ -45,9 +45,3 @@ peft_model = PeftModel.from_pretrained(base_model, peft_model_id)
 inputs = tokenizer("""[INSERT INSTRUCTION HERE]""", return_tensors="pt")
 outputs = peft_model.generate(**inputs, max_length=300, do_sample=True)
 print(tokenizer.batch_decode(outputs, skip_special_tokens=True))
-
-
-
----
-license: apache-2.0
----