Update README.md
Browse files
README.md
CHANGED
@@ -53,7 +53,7 @@ model = AutoModelForCausalLM.from_pretrained(
|
|
53 |
torch_dtype="auto",
|
54 |
trust_remote_code=True,
|
55 |
)
|
56 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
57 |
|
58 |
messages = [
|
59 |
{"role": "system", "content": "You are a helpful Emotional intelligence named as EMO-phi, remember to always answer users question in EMO style."},
|
@@ -67,7 +67,7 @@ pipe = pipeline(
|
|
67 |
)
|
68 |
|
69 |
generation_args = {
|
70 |
-
"max_new_tokens":
|
71 |
"return_full_text": False,
|
72 |
"temperature": 0.6,
|
73 |
"do_sample": True,
|
|
|
53 |
torch_dtype="auto",
|
54 |
trust_remote_code=True,
|
55 |
)
|
56 |
+
tokenizer = AutoTokenizer.from_pretrained("microsoft/Phi-3-mini-128k-instruct")
|
57 |
|
58 |
messages = [
|
59 |
{"role": "system", "content": "You are a helpful Emotional intelligence named as EMO-phi, remember to always answer users question in EMO style."},
|
|
|
67 |
)
|
68 |
|
69 |
generation_args = {
|
70 |
+
"max_new_tokens": 500,
|
71 |
"return_full_text": False,
|
72 |
"temperature": 0.6,
|
73 |
"do_sample": True,
|