set min width
Browse files
app.py
CHANGED
|
@@ -78,7 +78,7 @@ with demo:
|
|
| 78 |
with gr.TabItem("OPT: Open"):
|
| 79 |
plot_fn = functools.partial(get_plot, "OPT: Open Text Generation")
|
| 80 |
with gr.Row():
|
| 81 |
-
with gr.Column(scale=0.3):
|
| 82 |
gpu_selector = gpu_selector_fn(["3090", "T4", "T4 *2", "A100 (80GB)"])
|
| 83 |
omit_offload = omit_offload_fn()
|
| 84 |
gr.Markdown(
|
|
@@ -97,8 +97,7 @@ with demo:
|
|
| 97 |
"""
|
| 98 |
)
|
| 99 |
# Show plot when the gradio app is initialized
|
| 100 |
-
|
| 101 |
-
plot = gr.Image(value=plot_fn("A100 (80GB)", "No"))
|
| 102 |
# Update plot when any of the inputs change
|
| 103 |
plot_inputs = [gpu_selector, omit_offload]
|
| 104 |
gpu_selector.change(fn=plot_fn, inputs=plot_inputs, outputs=plot)
|
|
@@ -106,7 +105,7 @@ with demo:
|
|
| 106 |
with gr.TabItem("OPT: Summ"):
|
| 107 |
plot_fn = functools.partial(get_plot, "OPT: Summarization")
|
| 108 |
with gr.Row():
|
| 109 |
-
with gr.Column(scale=0.3):
|
| 110 |
gpu_selector = gpu_selector_fn(["3090", "T4", "T4 *2", "A100 (80GB)"])
|
| 111 |
omit_offload = omit_offload_fn()
|
| 112 |
gr.Markdown(
|
|
@@ -133,7 +132,7 @@ with demo:
|
|
| 133 |
with gr.TabItem("Whisper: ARS"):
|
| 134 |
plot_fn = functools.partial(get_plot, "Whisper: ARS")
|
| 135 |
with gr.Row():
|
| 136 |
-
with gr.Column(scale=0.3):
|
| 137 |
gpu_selector = gpu_selector_fn(["3090", "T4"])
|
| 138 |
omit_offload = omit_offload_fn()
|
| 139 |
gr.Markdown(
|
|
@@ -157,7 +156,7 @@ with demo:
|
|
| 157 |
with gr.TabItem("CodeGen: Code"):
|
| 158 |
plot_fn = functools.partial(get_plot, "CodeGen: Code Generation")
|
| 159 |
with gr.Row():
|
| 160 |
-
with gr.Column(scale=0.3):
|
| 161 |
gpu_selector = gpu_selector_fn(["3090", "T4", "T4 *2", "A100 (80GB)"])
|
| 162 |
omit_offload = omit_offload_fn()
|
| 163 |
gr.Markdown(
|
|
@@ -183,7 +182,7 @@ with demo:
|
|
| 183 |
with gr.TabItem("Flan-T5: Summ"):
|
| 184 |
plot_fn = functools.partial(get_plot, "Flan-T5: Summarization")
|
| 185 |
with gr.Row():
|
| 186 |
-
with gr.Column(scale=0.3):
|
| 187 |
gpu_selector = gpu_selector_fn(["3090", "T4", "T4 *2", "A100 (80GB)"])
|
| 188 |
omit_offload = omit_offload_fn()
|
| 189 |
gr.Markdown(
|
|
|
|
| 78 |
with gr.TabItem("OPT: Open"):
|
| 79 |
plot_fn = functools.partial(get_plot, "OPT: Open Text Generation")
|
| 80 |
with gr.Row():
|
| 81 |
+
with gr.Column(scale=0.3, min_width=200):
|
| 82 |
gpu_selector = gpu_selector_fn(["3090", "T4", "T4 *2", "A100 (80GB)"])
|
| 83 |
omit_offload = omit_offload_fn()
|
| 84 |
gr.Markdown(
|
|
|
|
| 97 |
"""
|
| 98 |
)
|
| 99 |
# Show plot when the gradio app is initialized
|
| 100 |
+
plot = gr.Image(value=plot_fn("A100 (80GB)", "No"))
|
|
|
|
| 101 |
# Update plot when any of the inputs change
|
| 102 |
plot_inputs = [gpu_selector, omit_offload]
|
| 103 |
gpu_selector.change(fn=plot_fn, inputs=plot_inputs, outputs=plot)
|
|
|
|
| 105 |
with gr.TabItem("OPT: Summ"):
|
| 106 |
plot_fn = functools.partial(get_plot, "OPT: Summarization")
|
| 107 |
with gr.Row():
|
| 108 |
+
with gr.Column(scale=0.3, min_width=200):
|
| 109 |
gpu_selector = gpu_selector_fn(["3090", "T4", "T4 *2", "A100 (80GB)"])
|
| 110 |
omit_offload = omit_offload_fn()
|
| 111 |
gr.Markdown(
|
|
|
|
| 132 |
with gr.TabItem("Whisper: ARS"):
|
| 133 |
plot_fn = functools.partial(get_plot, "Whisper: ARS")
|
| 134 |
with gr.Row():
|
| 135 |
+
with gr.Column(scale=0.3, min_width=200):
|
| 136 |
gpu_selector = gpu_selector_fn(["3090", "T4"])
|
| 137 |
omit_offload = omit_offload_fn()
|
| 138 |
gr.Markdown(
|
|
|
|
| 156 |
with gr.TabItem("CodeGen: Code"):
|
| 157 |
plot_fn = functools.partial(get_plot, "CodeGen: Code Generation")
|
| 158 |
with gr.Row():
|
| 159 |
+
with gr.Column(scale=0.3, min_width=200):
|
| 160 |
gpu_selector = gpu_selector_fn(["3090", "T4", "T4 *2", "A100 (80GB)"])
|
| 161 |
omit_offload = omit_offload_fn()
|
| 162 |
gr.Markdown(
|
|
|
|
| 182 |
with gr.TabItem("Flan-T5: Summ"):
|
| 183 |
plot_fn = functools.partial(get_plot, "Flan-T5: Summarization")
|
| 184 |
with gr.Row():
|
| 185 |
+
with gr.Column(scale=0.3, min_width=200):
|
| 186 |
gpu_selector = gpu_selector_fn(["3090", "T4", "T4 *2", "A100 (80GB)"])
|
| 187 |
omit_offload = omit_offload_fn()
|
| 188 |
gr.Markdown(
|