single col for blog post?
Browse files
app.py
CHANGED
@@ -78,26 +78,25 @@ with demo:
|
|
78 |
with gr.TabItem("OPT: Open"):
|
79 |
plot_fn = functools.partial(get_plot, "OPT: Open Text Generation")
|
80 |
with gr.Row():
|
81 |
-
|
82 |
-
|
83 |
-
omit_offload = omit_offload_fn()
|
84 |
-
gr.Markdown(
|
85 |
-
"""
|
86 |
-
### Assistant Model
|
87 |
-
- `facebook/opt-125m`
|
88 |
-
|
89 |
-
### Model Names:
|
90 |
-
- 1.3B: `facebook/opt-1.3b`
|
91 |
-
- 6.7B: `facebook/opt-6.7b`
|
92 |
-
- 30B: `facebook/opt-30b`
|
93 |
-
- 66B: `facebook/opt-66b`
|
94 |
-
|
95 |
-
### Dataset used as input prompt:
|
96 |
-
- C4 (en, validation set)
|
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)
|
|
|
78 |
with gr.TabItem("OPT: Open"):
|
79 |
plot_fn = functools.partial(get_plot, "OPT: Open Text Generation")
|
80 |
with gr.Row():
|
81 |
+
gpu_selector = gpu_selector_fn(["3090", "T4", "T4 *2", "A100 (80GB)"])
|
82 |
+
omit_offload = omit_offload_fn()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
83 |
# Show plot when the gradio app is initialized
|
84 |
plot = gr.Image(value=plot_fn("A100 (80GB)", "No"))
|
85 |
+
gr.Markdown(
|
86 |
+
"""
|
87 |
+
### Assistant Model
|
88 |
+
- `facebook/opt-125m`
|
89 |
+
|
90 |
+
### Model Names:
|
91 |
+
- 1.3B: `facebook/opt-1.3b`
|
92 |
+
- 6.7B: `facebook/opt-6.7b`
|
93 |
+
- 30B: `facebook/opt-30b`
|
94 |
+
- 66B: `facebook/opt-66b`
|
95 |
+
|
96 |
+
### Dataset used as input prompt:
|
97 |
+
- C4 (en, validation set)
|
98 |
+
"""
|
99 |
+
)
|
100 |
# Update plot when any of the inputs change
|
101 |
plot_inputs = [gpu_selector, omit_offload]
|
102 |
gpu_selector.change(fn=plot_fn, inputs=plot_inputs, outputs=plot)
|