Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Update .env.local.template
Browse files- .env.local.template +5 -7
.env.local.template
CHANGED
@@ -69,8 +69,7 @@ MODELS=`[
|
|
69 |
"parameters": {
|
70 |
"temperature": 0.4,
|
71 |
"repetition_penalty": 2,
|
72 |
-
"max_new_tokens":
|
73 |
-
"stop":["<|im_end|>"]
|
74 |
},
|
75 |
"extraBody":{
|
76 |
"presence_penalty":-0.9
|
@@ -99,17 +98,16 @@ MODELS=`[
|
|
99 |
}
|
100 |
],
|
101 |
"parameters": {
|
102 |
-
"temperature": 0.
|
103 |
"repetition_penalty": 2,
|
104 |
-
"max_new_tokens":
|
105 |
-
"stop":["<|im_end|>"]
|
106 |
},
|
107 |
"extraBody":{
|
108 |
-
"presence_penalty"
|
109 |
}
|
110 |
}
|
111 |
]`
|
112 |
-
#Repetition penality = presence_penalty + 2
|
113 |
|
114 |
TASK_MODEL=`{
|
115 |
"name": "czearing/article-title-generator",
|
|
|
69 |
"parameters": {
|
70 |
"temperature": 0.4,
|
71 |
"repetition_penalty": 2,
|
72 |
+
"max_new_tokens": 2000,
|
|
|
73 |
},
|
74 |
"extraBody":{
|
75 |
"presence_penalty":-0.9
|
|
|
98 |
}
|
99 |
],
|
100 |
"parameters": {
|
101 |
+
"temperature": 0.3,
|
102 |
"repetition_penalty": 2,
|
103 |
+
"max_new_tokens": 4000,
|
|
|
104 |
},
|
105 |
"extraBody":{
|
106 |
+
"presence_penalty":2
|
107 |
}
|
108 |
}
|
109 |
]`
|
110 |
+
#Repetition penality = presence_penalty + 2 # FOR TGI
|
111 |
|
112 |
TASK_MODEL=`{
|
113 |
"name": "czearing/article-title-generator",
|