bstraehle commited on
Commit
2411c9e
1 Parent(s): 4ac20b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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()