rzanoli commited on
Commit
e69ebae
·
1 Parent(s): b92514e

Small changes to preporcess vision model files

Browse files
Files changed (1) hide show
  1. preprocess_models_output.py +1 -1
preprocess_models_output.py CHANGED
@@ -122,7 +122,7 @@ def extract_data_from_file(file_path):
122
  continue
123
 
124
  # Extracts model configuration details
125
- if line.startswith("hf (pretrained="):
126
  start = line.find("pretrained=") + len("pretrained=")
127
  end = line.find(",", start)
128
  pretrained_model = line[start:end]
 
122
  continue
123
 
124
  # Extracts model configuration details
125
+ if line.startswith("hf (pretrained=") or line.startswith("hf-multimodal (pretrained="):
126
  start = line.find("pretrained=") + len("pretrained=")
127
  end = line.find(",", start)
128
  pretrained_model = line[start:end]