Spaces:
Running
Running
Refactor model configuration to use ACTIVE_MODEL for improved clarity and consistency
Browse files
app.py
CHANGED
|
@@ -273,7 +273,8 @@ with gr.Blocks() as demo:
|
|
| 273 |
with gr.Column(scale=3):
|
| 274 |
chatbot = gr.Chatbot(
|
| 275 |
label="Chat",
|
| 276 |
-
avatar_images=None
|
|
|
|
| 277 |
)
|
| 278 |
|
| 279 |
with gr.Row():
|
|
@@ -383,18 +384,18 @@ with gr.Blocks() as demo:
|
|
| 383 |
gr.Markdown(f"""
|
| 384 |
**Base Model Path:**
|
| 385 |
```
|
| 386 |
-
{
|
| 387 |
```
|
| 388 |
|
| 389 |
**Fine-tuned Model Path:**
|
| 390 |
```
|
| 391 |
-
{
|
| 392 |
```
|
| 393 |
|
| 394 |
**LoRA Configuration:**
|
| 395 |
-
- Rank (r): {
|
| 396 |
-
- Alpha: {
|
| 397 |
-
- Dropout: {
|
| 398 |
""")
|
| 399 |
|
| 400 |
with gr.Tab("Model Training"):
|
|
|
|
| 273 |
with gr.Column(scale=3):
|
| 274 |
chatbot = gr.Chatbot(
|
| 275 |
label="Chat",
|
| 276 |
+
avatar_images=None,
|
| 277 |
+
type='messages'
|
| 278 |
)
|
| 279 |
|
| 280 |
with gr.Row():
|
|
|
|
| 384 |
gr.Markdown(f"""
|
| 385 |
**Base Model Path:**
|
| 386 |
```
|
| 387 |
+
{ACTIVE_MODEL['training']['base_model_path']}
|
| 388 |
```
|
| 389 |
|
| 390 |
**Fine-tuned Model Path:**
|
| 391 |
```
|
| 392 |
+
{ACTIVE_MODEL['training']['fine_tuned_path']}
|
| 393 |
```
|
| 394 |
|
| 395 |
**LoRA Configuration:**
|
| 396 |
+
- Rank (r): {ACTIVE_MODEL['training']['lora_config']['r']}
|
| 397 |
+
- Alpha: {ACTIVE_MODEL['training']['lora_config']['lora_alpha']}
|
| 398 |
+
- Dropout: {ACTIVE_MODEL['training']['lora_config']['lora_dropout']}
|
| 399 |
""")
|
| 400 |
|
| 401 |
with gr.Tab("Model Training"):
|