github
/
InternLM
mirror of https://github.com/InternLM/InternLM
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Update passkey_retrieval.py

pull/426/head
tpoisonooo 2023-10-19 16:30:22 +08:00 committed by GitHub
parent d2cc5f61e3
commit 69e0d63b33
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
tools/passkey_retrieval.py
View File

@ -2,7 +2,6 @@ import argparse
import random import random
from numpy import random from numpy import random
from transformers import AutoTokenizer, AutoModelForCausalLM from transformers import AutoTokenizer, AutoModelForCausalLM
import pdb
def parse_config(): def parse_config():
parser = argparse.ArgumentParser(description='arg parser') parser = argparse.ArgumentParser(description='arg parser')