Small changes to preporcess vision model files
Browse files
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]
|