oobabooga commited on
Commit
2c659e0
·
verified ·
1 Parent(s): 45ce836

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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'):