Steven10429 commited on
Commit
47601b7
·
verified ·
1 Parent(s): 0439ea7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -196,11 +196,12 @@ def quantize_and_push_model(model_path, repo_id, bits=8):
196
  def process_model(base_model, lora_model, repo_name, hf_token, progress=gr.Progress()):
197
  """处理模型的主函数,用于Gradio界面"""
198
  try:
199
- login(hf_token)
200
  os.environ["HF_TOKEN"] = hf_token
201
  api = HfApi(token=hf_token)
 
202
  if repo_name == "Auto":
203
- repo_name = base_model.split("/")[-1] + "_" + lora_model.split("/")[-1]
204
 
205
  # 清空之前的日志
206
  current_logs.clear()
 
196
  def process_model(base_model, lora_model, repo_name, hf_token, progress=gr.Progress()):
197
  """处理模型的主函数,用于Gradio界面"""
198
  try:
199
+ login(hf_token) # 我不理解为什么登录一次不行,非得放到环境变量里
200
  os.environ["HF_TOKEN"] = hf_token
201
  api = HfApi(token=hf_token)
202
+ username = api.whoami["name"]
203
  if repo_name == "Auto":
204
+ repo_name = username + "/" + base_model.split("/")[-1] + "_" + lora_model.split("/")[-1]
205
 
206
  # 清空之前的日志
207
  current_logs.clear()