yash-srivastava19 commited on
Commit
22a629d
·
verified ·
1 Parent(s): 78f9f8d

Update custom_llm.py

Browse files

Some subtle changes from lagacy to new API

Files changed (1) hide show
  1. custom_llm.py +1 -1
custom_llm.py CHANGED
@@ -18,7 +18,7 @@ class CustomLLM(LLM):
18
  raise ValueError("stop kwargs are not permitted.")
19
 
20
  response = co.chat(message=f'{prompt}', max_tokens=1700, temperature=0.9)
21
- print(response.generations[0].text)
22
 
23
  return f' ```python \n {response.generations[0].text} \n ``` '
24
 
 
18
  raise ValueError("stop kwargs are not permitted.")
19
 
20
  response = co.chat(message=f'{prompt}', max_tokens=1700, temperature=0.9)
21
+ print(response.text)
22
 
23
  return f' ```python \n {response.generations[0].text} \n ``` '
24