Spaces:

togethercomputer
/

GPT-JT

Runtime error

jrb commited on May 9, 2023

Commit

af1594d

•

1 Parent(s): 036a7be

update model names

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ def infer(
     stop="\n"
 ):
     model_name_map = {
-        "GPT-JT-6B-v1": "Together-gpt-JT-6B-v1",
     }
     max_new_tokens = int(max_new_tokens)
     num_completions = int(num_completions)
@@ -45,7 +45,7 @@ def infer(
     if prompt=="":
         prompt = " "
     my_post_dict = {
-        "model": "Together-gpt-JT-6B-v1",
         "prompt": prompt,
         "top_p": top_p,
         "top_k": top_k,

     stop="\n"
 ):
     model_name_map = {
+        "GPT-JT-6B-v1": "togethercomputer/GPT-JT-6B-v1",
     }
     max_new_tokens = int(max_new_tokens)
     num_completions = int(num_completions)
     if prompt=="":
         prompt = " "
     my_post_dict = {
+        "model": "togethercomputer/GPT-JT-6B-v1",
         "prompt": prompt,
         "top_p": top_p,
         "top_k": top_k,