Update app.py
Browse files
app.py
CHANGED
@@ -1,4 +1,4 @@
|
|
1 |
-
# Run on Google TPU v5e 2x4 or equivalent (220 vCPU, 380 GB RAM, 128 GB VRAM)
|
2 |
|
3 |
import gradio as gr
|
4 |
import os, torch
|
@@ -149,6 +149,6 @@ demo = gr.Interface(fn=process,
|
|
149 |
gr.Textbox(label = "User Prompt", value = USER_PROMPT, lines = 2),
|
150 |
gr.Textbox(label = "SQL Context", value = SQL_CONTEXT, lines = 4)],
|
151 |
outputs=[gr.Textbox(label = "Prompt Completion", value = os.environ["OUTPUT"])],
|
152 |
-
title = "Supervised Fine-Tuning",
|
153 |
description = os.environ["DESCRIPTION"])
|
154 |
demo.launch()
|
|
|
1 |
+
# Run full fine-tuning on Google TPU v5e 2x4 or equivalent (220 vCPU, 380 GB RAM, 128 GB VRAM)
|
2 |
|
3 |
import gradio as gr
|
4 |
import os, torch
|
|
|
149 |
gr.Textbox(label = "User Prompt", value = USER_PROMPT, lines = 2),
|
150 |
gr.Textbox(label = "SQL Context", value = SQL_CONTEXT, lines = 4)],
|
151 |
outputs=[gr.Textbox(label = "Prompt Completion", value = os.environ["OUTPUT"])],
|
152 |
+
title = "Supervised Fine-Tuning (SFT)",
|
153 |
description = os.environ["DESCRIPTION"])
|
154 |
demo.launch()
|