Update app.py
Browse files
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()
|