nickmuchi commited on
Commit
87081eb
1 Parent(s): e0e0e4d

Update functions.py

Browse files
Files changed (1) hide show
  1. functions.py +1 -1
functions.py CHANGED
@@ -512,7 +512,7 @@ def generate_eval(raw_text, N, chunk):
512
  n = len(raw_text)
513
  starting_indices = [random.randint(0, n-chunk) for _ in range(N)]
514
  sub_sequences = [raw_text[i:i+chunk] for i in starting_indices]
515
- chain = QAGenerationChain.from_llm(ChatOpenAI(temperature=0))
516
  eval_set = []
517
  for i, b in enumerate(sub_sequences):
518
  try:
 
512
  n = len(raw_text)
513
  starting_indices = [random.randint(0, n-chunk) for _ in range(N)]
514
  sub_sequences = [raw_text[i:i+chunk] for i in starting_indices]
515
+ chain = QAGenerationChain.from_llm(ChatOpenAI(temperature=0, model_name='gpt-4-0613'))
516
  eval_set = []
517
  for i, b in enumerate(sub_sequences):
518
  try: