| import os | |
| import base64 | |
| def decode_filename(filename): | |
| name, ext = os.path.splitext(filename) | |
| try: | |
| decoded = base64.b64decode(name.encode('utf-8')).decode('utf-8') | |
| return f"{decoded}{ext}" | |
| except Exception as e: | |
| print(f"Skip: {filename} (cannot decode)") | |
| return None | |
| def main(): | |
| for fname in os.listdir('.'): | |
| if os.path.isfile(fname) and fname.endswith(".safetensors"): | |
| new_name = decode_filename(fname) | |
| if new_name: | |
| os.rename(fname, new_name) | |
| print(f"Restored: {fname} -> {new_name}") | |
| if __name__ == "__main__": | |
| main() | |