joaogante HF staff commited on
Commit
f74e135
·
1 Parent(s): a0b4352

fiddle with fig options

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -145,7 +145,7 @@ def get_plot(model_name, generate_type):
145
  df["framework"] = ["PyTorch", "TF (Eager Execition)", "TF (XLA)"]
146
  df = pd.melt(df, id_vars=["framework"], value_vars=["T4", "3090", "A100"])
147
 
148
- fig = plt.figure(figsize=(100, 6), dpi=200)
149
  g = sns.catplot(
150
  data=df, kind="bar",
151
  x="variable", y="value", hue="framework",
@@ -153,7 +153,6 @@ def get_plot(model_name, generate_type):
153
  )
154
  g.despine(left=True)
155
  g.set_axis_labels("GPU", "Generation time (ms)")
156
- g.set(title=f"{generate_type} generation time for {model_name} (no bar means OOM)")
157
  g.legend.set_title("Framework")
158
  return plt.gcf()
159
 
 
145
  df["framework"] = ["PyTorch", "TF (Eager Execition)", "TF (XLA)"]
146
  df = pd.melt(df, id_vars=["framework"], value_vars=["T4", "3090", "A100"])
147
 
148
+ # fig = plt.figure(figsize=(100, 6), dpi=200)
149
  g = sns.catplot(
150
  data=df, kind="bar",
151
  x="variable", y="value", hue="framework",
 
153
  )
154
  g.despine(left=True)
155
  g.set_axis_labels("GPU", "Generation time (ms)")
 
156
  g.legend.set_title("Framework")
157
  return plt.gcf()
158