diff --git a/util/convert_weights.py b/util/convert_weights.py index 187cd2f..bd6750a 100644 --- a/util/convert_weights.py +++ b/util/convert_weights.py @@ -22,12 +22,8 @@ import numpy as np import argparse import os -# WIP - DO NOT MERGE # Requires torch 2.2 and gemma package from https://github.com/google/gemma_pytorch -## parameters -## model, tokenizer, model type, - def check_file_exists(value): if not os.path.exists(str(value)): raise argparse.ArgumentTypeError("The file %s does not appear to exist." % value)