Sarah Ciston commited on
Commit
73dc4d4
·
1 Parent(s): c615898

try with community gpt2

Browse files
Files changed (2) hide show
  1. README.md +3 -3
  2. sketch.js +2 -2
README.md CHANGED
@@ -10,10 +10,10 @@ hf_oauth_scopes:
10
  - read-repos
11
  - write-repos
12
  models:
13
- - mistralai/Mistral-Nemo-Instruct-2407
14
  # - Xenova/distilgpt2
15
- # - HuggingFaceH4/zephyr-7b-beta
16
- # - openai-community/gpt2
17
  # - meta-llama/Meta-Llama-3-70B-Instruct
18
  # - Xenova/detr-resnet-50
19
  # - Xenova/gpt2
 
10
  - read-repos
11
  - write-repos
12
  models:
13
+ # - gpt-3.5-turbo
14
  # - Xenova/distilgpt2
15
+ - HuggingFaceH4/zephyr-7b-beta
16
+ - openai-community/gpt2
17
  # - meta-llama/Meta-Llama-3-70B-Instruct
18
  # - Xenova/detr-resnet-50
19
  # - Xenova/gpt2
sketch.js CHANGED
@@ -223,7 +223,7 @@ async function runModel(PREPROMPT, PROMPT){
223
  // inference API version, not working in spaces
224
 
225
  // let MODELNAME = 'mistralai/Mistral-Nemo-Instruct-2407'
226
- let MODELNAME = "mistralai/Mistral-7B-Instruct-v0.2"
227
 
228
  let out = await inference.chatCompletion({
229
  model: MODELNAME,
@@ -234,7 +234,7 @@ async function runModel(PREPROMPT, PROMPT){
234
  role: "user",
235
  content: PROMPT
236
  }],
237
- max_tokens: 100
238
  });
239
 
240
  console.log(out)
 
223
  // inference API version, not working in spaces
224
 
225
  // let MODELNAME = 'mistralai/Mistral-Nemo-Instruct-2407'
226
+ let MODELNAME = "openai-community/gpt2"
227
 
228
  let out = await inference.chatCompletion({
229
  model: MODELNAME,
 
234
  role: "user",
235
  content: PROMPT
236
  }],
237
+ max_new_tokens: 100
238
  });
239
 
240
  console.log(out)