joaogante HF staff commited on
Commit
dbdfda7
1 Parent(s): 24dcfda

single col for blog post?

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