brurei commited on
Commit
299af11
1 Parent(s): 73eb400

Update CD2AiTrainning.py

Browse files
Files changed (1) hide show
  1. CD2AiTrainning.py +1 -1
CD2AiTrainning.py CHANGED
@@ -16,7 +16,7 @@ def construct_index(directory_path):
16
 
17
  prompt_helper = PromptHelper(max_input_size, num_outputs, max_chunk_overlap, chunk_size_limit=chunk_size_limit)
18
 
19
- llm_predictor = LLMPredictor(llm=OpenAI(temperature=0.7, model_name="text-davinci-003", max_tokens=num_outputs))
20
 
21
  documents = SimpleDirectoryReader(directory_path).load_data()
22
 
 
16
 
17
  prompt_helper = PromptHelper(max_input_size, num_outputs, max_chunk_overlap, chunk_size_limit=chunk_size_limit)
18
 
19
+ llm_predictor = LLMPredictor(llm=OpenAI(temperature=0.7, model_name="gpt-3.5-turbo-instruct", max_tokens=num_outputs))
20
 
21
  documents = SimpleDirectoryReader(directory_path).load_data()
22