Spaces:
Sleeping
Sleeping
Update custom_llm.py
Browse filesSome subtle changes from lagacy to new API
- 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.
|
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 |
|