parameter viewer

app.py

@@ -67,7 +67,7 @@ subtotal_operations = [
 
 
 
-col1, col2, col3, col4, col5 = st.columns([1,1.5,2.5,2.5,0.1])
+col1, col2, col3, col4, col5 = st.columns([0.8, 2, 2.5, 2.5, 0.01])
 
 inference_config = {}
 parameter_count = {}
@@ -141,6 +141,17 @@ with col2:
     st.write(f'model_total_size (Byte): {model_total_size_in_byte:,}')
 
 
+    # add parameter viewer
+    if model_config['model']:
+        header4("Parameters Viewer")
+        weight_generic = st.selectbox('Select weight:', options=model_config['module_classes'])
+        modules = {}
+        for module in model_config['module_classes'][weight_generic]:
+            modules.update(module)
+        modules = {k: list(v) for k, v in modules.items()}
+        modules = pd.DataFrame(list(modules.items()), columns=["Parameter", "Shape"])
+        st.markdown(create_table(modules))
+
 with col3: # Prefilling
     prefilling_operation_count = prefilling_operation(model_config, inference_config)
     prefilling_activation_memory_count = prefilling_activation_memory(model_config, inference_config)