not-lain commited on
Commit
64329dd
·
verified ·
1 Parent(s): 360678e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -105,7 +105,7 @@ def setup_submission(system_prompt="", history=[], chat_format="sharegpt"):
105
  return history
106
 
107
 
108
- def save_sft_data(system_prompt="", history=[], sft_chat_format="sharegpt", oauth_token: gr.OAuthToken | None):
109
  """
110
  A function that pushes the data to the hub.
111
  """
@@ -142,7 +142,7 @@ def save_sft_data(system_prompt="", history=[], sft_chat_format="sharegpt", oaut
142
 
143
 
144
  def save_dpo_data(
145
- system_prompt="", history=[], chosen="", rejected="", dpo_chat_format="sharegpt", oauth_token: gr.OAuthToken | None
146
  ):
147
  """
148
  A function that pushes the data to the hub.
 
105
  return history
106
 
107
 
108
+ def save_sft_data(system_prompt="", history=[], sft_chat_format="sharegpt", oauth_token: gr.OAuthToken | None = None):
109
  """
110
  A function that pushes the data to the hub.
111
  """
 
142
 
143
 
144
  def save_dpo_data(
145
+ system_prompt="", history=[], chosen="", rejected="", dpo_chat_format="sharegpt", oauth_token: gr.OAuthToken | None = None
146
  ):
147
  """
148
  A function that pushes the data to the hub.