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()
|