Update app.py
Browse files
app.py
CHANGED
@@ -103,7 +103,7 @@ def load_metadata_from_file(file_obj):
|
|
103 |
if key.endswith('.block_count'):
|
104 |
extracted_fields['n_layers'] = value
|
105 |
elif key.endswith('.attention.head_count_kv'):
|
106 |
-
extracted_fields['n_kv_heads'] = value
|
107 |
elif key.endswith('.embedding_length'):
|
108 |
extracted_fields['embedding_dim'] = value
|
109 |
elif key.endswith('.context_length'):
|
|
|
103 |
if key.endswith('.block_count'):
|
104 |
extracted_fields['n_layers'] = value
|
105 |
elif key.endswith('.attention.head_count_kv'):
|
106 |
+
extracted_fields['n_kv_heads'] = max(value) if isinstance(value, list) else value
|
107 |
elif key.endswith('.embedding_length'):
|
108 |
extracted_fields['embedding_dim'] = value
|
109 |
elif key.endswith('.context_length'):
|