Spaces:
Running
Running
update
Browse files- plot_results.py +6 -0
plot_results.py
CHANGED
@@ -133,6 +133,12 @@ def create_performance_plot(csv_path='benchmark_results.csv', metadata_path='met
|
|
133 |
frontier_x, frontier_y = zip(*frontier_points)
|
134 |
plt.plot(frontier_x, frontier_y, 'r--', label='Pareto frontier')
|
135 |
|
|
|
|
|
|
|
|
|
|
|
|
|
136 |
# Customize the plot
|
137 |
plt.grid(True, linestyle='--', alpha=0.7)
|
138 |
plt.xlabel('Model Size (billions of parameters)')
|
|
|
133 |
frontier_x, frontier_y = zip(*frontier_points)
|
134 |
plt.plot(frontier_x, frontier_y, 'r--', label='Pareto frontier')
|
135 |
|
136 |
+
# Add vertical line for consumer GPU budget
|
137 |
+
plt.axvline(x=12, color='gray', linestyle=':', label='Consumer-budget GPU limit', ymin=-0.15, clip_on=False)
|
138 |
+
plt.text(12, -0.15, 'Consumer-budget\nGPU (24GB) limit\nin full precision',
|
139 |
+
horizontalalignment='center', verticalalignment='top',
|
140 |
+
transform=plt.gca().get_xaxis_transform())
|
141 |
+
|
142 |
# Customize the plot
|
143 |
plt.grid(True, linestyle='--', alpha=0.7)
|
144 |
plt.xlabel('Model Size (billions of parameters)')
|