Sebastien De Greef
commited on
Commit
•
c0f4fad
1
Parent(s):
457af58
Add auth button
Browse files- app.py +1 -0
- requirements.txt +1 -0
app.py
CHANGED
@@ -149,6 +149,7 @@ def save_model(model_name, hub_model_name, hub_token, gguf_16bit, gguf_8bit, ggu
|
|
149 |
with gr.Blocks(title="Unsloth fine-tuning") as demo:
|
150 |
with gr.Column():
|
151 |
gr.Image("unsloth.png", width="300px", interactive=False, show_download_button=False, show_label=False)
|
|
|
152 |
with gr.Column():
|
153 |
gr.Markdown(f"**User:** {hf_user}\n\n**GPU Information:** {gpu_stats.name} ({max_memory} GB)\n\n[Unsloth Docs](http://docs.unsloth.com/)\n\n[Unsloth GitHub](https://github.com/unslothai/unsloth)")
|
154 |
with gr.Tab("Base Model Parameters"):
|
|
|
149 |
with gr.Blocks(title="Unsloth fine-tuning") as demo:
|
150 |
with gr.Column():
|
151 |
gr.Image("unsloth.png", width="300px", interactive=False, show_download_button=False, show_label=False)
|
152 |
+
gr.LoginButton()
|
153 |
with gr.Column():
|
154 |
gr.Markdown(f"**User:** {hf_user}\n\n**GPU Information:** {gpu_stats.name} ({max_memory} GB)\n\n[Unsloth Docs](http://docs.unsloth.com/)\n\n[Unsloth GitHub](https://github.com/unslothai/unsloth)")
|
155 |
with gr.Tab("Base Model Parameters"):
|
requirements.txt
CHANGED
@@ -5,4 +5,5 @@ peft
|
|
5 |
accelerate
|
6 |
bitsandbytes
|
7 |
gradio
|
|
|
8 |
tensorboard
|
|
|
5 |
accelerate
|
6 |
bitsandbytes
|
7 |
gradio
|
8 |
+
gradio[oauth]
|
9 |
tensorboard
|