Spaces:
Running
Running
Update merge.py
Browse files
merge.py
CHANGED
@@ -87,7 +87,7 @@ def merge_folder(tensor_map, directory_path, p, lambda_val):
|
|
87 |
raise "Could not find model files"
|
88 |
|
89 |
for filename in os.listdir(directory_path):
|
90 |
-
file_path = os.path.join(directory_path, filename)
|
91 |
f, keys2 = read_tensors(file_path, ext)
|
92 |
if keys2:
|
93 |
common_keys = keys1.intersection(keys2)
|
|
|
87 |
raise "Could not find model files"
|
88 |
|
89 |
for filename in os.listdir(directory_path):
|
90 |
+
file_path = os.path.join(directory_path, filename)
|
91 |
f, keys2 = read_tensors(file_path, ext)
|
92 |
if keys2:
|
93 |
common_keys = keys1.intersection(keys2)
|