mirror of https://github.com/InternLM/InternLM
Update passkey_retrieval.py
parent
d2cc5f61e3
commit
69e0d63b33
|
@ -2,7 +2,6 @@ import argparse
|
|||
import random
|
||||
from numpy import random
|
||||
from transformers import AutoTokenizer, AutoModelForCausalLM
|
||||
import pdb
|
||||
|
||||
def parse_config():
|
||||
parser = argparse.ArgumentParser(description='arg parser')
|
||||
|
|
Loading…
Reference in New Issue