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'):
|