Spaces:
Sleeping
Sleeping
JoFrost
commited on
Commit
·
4f5033f
1
Parent(s):
6e8146a
chore: flan as default model
Browse files
.env
CHANGED
|
@@ -30,10 +30,10 @@ REJECT_UNAUTHORIZED=true
|
|
| 30 |
# 'name', 'userMessageToken', 'assistantMessageToken' are required
|
| 31 |
MODELS=`[
|
| 32 |
{
|
| 33 |
-
"name": "
|
| 34 |
"is_local": true,
|
| 35 |
-
"
|
| 36 |
-
"type": "
|
| 37 |
"userMessageToken": "<|prompter|>",
|
| 38 |
"assistantMessageToken": "<|assistant|>",
|
| 39 |
"messageEndToken": "</s>",
|
|
@@ -41,15 +41,17 @@ MODELS=`[
|
|
| 41 |
"promptExamples": [
|
| 42 |
{
|
| 43 |
"title": "What is love?",
|
| 44 |
-
"prompt": "
|
| 45 |
-
},
|
| 46 |
-
|
|
|
|
|
|
|
| 47 |
"title": "What does end-to-end protected mean?",
|
| 48 |
-
"prompt": "
|
| 49 |
}
|
| 50 |
],
|
| 51 |
"parameters": {
|
| 52 |
-
"temperature": 0.
|
| 53 |
"top_p": 0.95,
|
| 54 |
"repetition_penalty": 1.2,
|
| 55 |
"top_k": 50,
|
|
@@ -58,10 +60,10 @@ MODELS=`[
|
|
| 58 |
}
|
| 59 |
},
|
| 60 |
{
|
| 61 |
-
"name": "
|
| 62 |
"is_local": true,
|
| 63 |
-
"
|
| 64 |
-
"type": "
|
| 65 |
"userMessageToken": "<|prompter|>",
|
| 66 |
"assistantMessageToken": "<|assistant|>",
|
| 67 |
"messageEndToken": "</s>",
|
|
@@ -69,17 +71,15 @@ MODELS=`[
|
|
| 69 |
"promptExamples": [
|
| 70 |
{
|
| 71 |
"title": "What is love?",
|
| 72 |
-
"prompt": "
|
| 73 |
-
},
|
| 74 |
-
|
| 75 |
-
"prompt": "Who is Ada Lovelace?"
|
| 76 |
-
}, {
|
| 77 |
"title": "What does end-to-end protected mean?",
|
| 78 |
-
"prompt": "
|
| 79 |
}
|
| 80 |
],
|
| 81 |
"parameters": {
|
| 82 |
-
"temperature": 0.
|
| 83 |
"top_p": 0.95,
|
| 84 |
"repetition_penalty": 1.2,
|
| 85 |
"top_k": 50,
|
|
|
|
| 30 |
# 'name', 'userMessageToken', 'assistantMessageToken' are required
|
| 31 |
MODELS=`[
|
| 32 |
{
|
| 33 |
+
"name": "Xenova/LaMini-Flan-T5-783M",
|
| 34 |
"is_local": true,
|
| 35 |
+
"is_code": false,
|
| 36 |
+
"type": "text2text-generation",
|
| 37 |
"userMessageToken": "<|prompter|>",
|
| 38 |
"assistantMessageToken": "<|assistant|>",
|
| 39 |
"messageEndToken": "</s>",
|
|
|
|
| 41 |
"promptExamples": [
|
| 42 |
{
|
| 43 |
"title": "What is love?",
|
| 44 |
+
"prompt": "What is love?"
|
| 45 |
+
}, {
|
| 46 |
+
"title": "Who is Ada Lovelace?",
|
| 47 |
+
"prompt": "Who is Ada Lovelace?"
|
| 48 |
+
}, {
|
| 49 |
"title": "What does end-to-end protected mean?",
|
| 50 |
+
"prompt": "What does end-to-end protected mean?"
|
| 51 |
}
|
| 52 |
],
|
| 53 |
"parameters": {
|
| 54 |
+
"temperature": 0.8,
|
| 55 |
"top_p": 0.95,
|
| 56 |
"repetition_penalty": 1.2,
|
| 57 |
"top_k": 50,
|
|
|
|
| 60 |
}
|
| 61 |
},
|
| 62 |
{
|
| 63 |
+
"name": "microsoft/phi-1_5",
|
| 64 |
"is_local": true,
|
| 65 |
+
"is_phi": true,
|
| 66 |
+
"type": "text-generation",
|
| 67 |
"userMessageToken": "<|prompter|>",
|
| 68 |
"assistantMessageToken": "<|assistant|>",
|
| 69 |
"messageEndToken": "</s>",
|
|
|
|
| 71 |
"promptExamples": [
|
| 72 |
{
|
| 73 |
"title": "What is love?",
|
| 74 |
+
"prompt": "Describe what love is in a few sentences."
|
| 75 |
+
},
|
| 76 |
+
{
|
|
|
|
|
|
|
| 77 |
"title": "What does end-to-end protected mean?",
|
| 78 |
+
"prompt": "Describe what does end-to-end protected mean in a few sentences."
|
| 79 |
}
|
| 80 |
],
|
| 81 |
"parameters": {
|
| 82 |
+
"temperature": 0.5,
|
| 83 |
"top_p": 0.95,
|
| 84 |
"repetition_penalty": 1.2,
|
| 85 |
"top_k": 50,
|