Spaces:

chansung
/

vid2persona

Paused

chansung commited on Mar 11

Commit

3777a20

•

1 Parent(s): 8b86a0a

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -40,7 +40,7 @@ async def conversation(
         prompt_tpl_path, model_id,
         max_input_token_length, max_new_tokens,
         temperature, top_p, top_k,
-        repetition_penalty, hf_token=None#init.hf_access_token
     ):
         last_message = messages[-1]
         last_message[1] = last_message[1] + partial_response
@@ -66,7 +66,7 @@ async def regen_conversation(
             prompt_tpl_path, model_id,
             max_input_token_length, max_new_tokens,
             temperature, top_p, top_k,
-            repetition_penalty, hf_token=None#init.hf_access_token
         ):
             last_message = messages[-1]
             last_message[1] = last_message[1] + partial_response

         prompt_tpl_path, model_id,
         max_input_token_length, max_new_tokens,
         temperature, top_p, top_k,
+        repetition_penalty, hf_token=init.hf_access_token
     ):
         last_message = messages[-1]
         last_message[1] = last_message[1] + partial_response
             prompt_tpl_path, model_id,
             max_input_token_length, max_new_tokens,
             temperature, top_p, top_k,
+            repetition_penalty, hf_token=init.hf_access_token
         ):
             last_message = messages[-1]
             last_message[1] = last_message[1] + partial_response