happyz
/
gemma.cpp
mirror of https://github.com/google/gemma.cpp.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Remove WIP messages.

This commit is contained in:
pculliton 2024-03-13 11:36:19 -04:00 committed by GitHub
parent b6831a2256
commit f520e5c25c
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
util/convert_weights.py
View File

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