diff --git a/app.py b/app.py index e3ecaa8..6493906 100644 --- a/app.py +++ b/app.py @@ -145,10 +145,10 @@ def move_to_second_view_from_tb(tb, evt: gr.SelectData): info["thumb"], f"## {selected_model}", f"**Parameters**\n: Approx. {info['parameters']}", - f"**🤗 Hub(base)**\n: {info['hub(base)']}", - f"**🤗 Hub(LoRA)**\n: {info['hub(ckpt)']}", - f"**🤗 Hub(GPTQ)**\n: {info['hub(gptq)']}", - f"**🤗 Hub(GPTQ_BASE)**\n: {info['hub(gptq_base)']}", + f"**Hugging Face Hub(base)**\n: {info['hub(base)']}", + f"**Hugging Face Hub(LoRA)**\n: {info['hub(ckpt)']}", + f"**Hugging Face Hub(GPTQ)**\n: {info['hub(gptq)']}", + f"**Hugging Face Hub(GPTQ_BASE)**\n: {info['hub(gptq_base)']}", info['desc'], f"""**Min VRAM requirements** : | half precision | load_in_8bit | load_in_4bit | @@ -415,10 +415,10 @@ def move_to_second_view(btn): info["thumb"], f"## {btn}", f"**Parameters**\n: Approx. {info['parameters']}", - f"**🤗 Hub(base)**\n: {info['hub(base)']}", - f"**🤗 Hub(LoRA)**\n: {info['hub(ckpt)']}", - f"**🤗 Hub(GPTQ)**\n: {info['hub(gptq)']}", - f"**🤗 Hub(GPTQ_BASE)**\n: {info['hub(gptq_base)']}", + f"**Hugging Face Hub(base)**\n: {info['hub(base)']}", + f"**Hugging Face Hub(LoRA)**\n: {info['hub(ckpt)']}", + f"**Hugging Face Hub(GPTQ)**\n: {info['hub(gptq)']}", + f"**Hugging Face Hub(GPTQ_BASE)**\n: {info['hub(gptq_base)']}", info['desc'], f"""**Min VRAM requirements** : | half precision | load_in_8bit | load_in_4bit | @@ -459,14 +459,15 @@ def download_completed( ): global local_files_only - print(model_gptq_base) + print(f"model_name: {model_name}") + print(f"model_base: {model_base}") tmp_args = types.SimpleNamespace() - tmp_args.model_name = model_name[4:-6] - tmp_args.base_url = model_base.split(":")[-1].split("