Update README.md

README.md

@@ -5,7 +5,7 @@ library_name: transformers
 pipeline_tag: text-generation
 license: mit
 base_model:
-- meta-llama/Llama-2-7b
+- pinkmanlove/llama-7b-hf
 ---
 
 
@@ -36,7 +36,7 @@ Yohan Na*, Dahye Kim*, and Dong-Kyu chae.
 <br>
 
 ## Introduction
-DialRet is a dialogue-specific language model designed for multi-session conversations. 
+DialRet is a dialogue-specific language model designed for multi-session conversations. (base model: llama-1-7b)
 Instead of using memory modules, it leverages long-context LMs and instruction-tuning across eight tasks (e.g., dialogue generation, summarization, speaker relation extraction). 
 It improves understanding and retention of past dialogues.