Fixing full spectrum.

app.py

@@ -4,6 +4,10 @@ import numpy as np
 import pandas as pd
 import re
 import torch
+import altair as alt
+
+
+alt.data_transformers.disable_max_rows()
 
 number_re = re.compile(r"\.[0-9]*\.")
 
@@ -25,10 +29,10 @@ def scatter_plot_fn(group_name):
 
 
 def find_choices(state_dict):
+    if not state_dict:
+        return []
     global DATA
-    layered_tensors = [
-        k for k, v in state_dict.items() if number_re.findall(k) and len(v.shape) == 2
-    ]
+    layered_tensors = [k for k, v in state_dict.items() if number_re.findall(k) and len(v.shape) == 2]
     choices = set()
     data = []
     for name in layered_tensors:
@@ -38,7 +42,7 @@ def find_choices(state_dict):
 
         svdvals = torch.linalg.svdvals(state_dict[name])
         svdvals /= svdvals.sum()
-        for rank, val in enumerate(svdvals.tolist()[:300]):
+        for rank, val in enumerate(svdvals.tolist()):
             data.append((name, layer, group_name, rank, val))
     data = np.array(data)
     DATA = pd.DataFrame(data, columns=["name", "layer", "group_name", "rank", "val"])
@@ -71,4 +75,4 @@ with gr.Blocks() as scatter_plot:
     weights.change(fn=scatter_plot_fn, inputs=weights, outputs=plot)
 
 if __name__ == "__main__":
-    scatter_plot.launch()
+    scatter_plot.launch(share=True)