mirror of
https://github.com/ollama/ollama-python.git
synced 2026-09-25 03:37:24 +00:00
royjhan
33c4b61ff9
* add suffix * update fill-in-the-middle example * keep example * lint * variables
23 lines
346 B
Python
23 lines
346 B
Python
from ollama import generate
|
|
|
|
prompt = '''def remove_non_ascii(s: str) -> str:
|
|
""" '''
|
|
|
|
suffix = """
|
|
return result
|
|
"""
|
|
|
|
response = generate(
|
|
model='codellama:7b-code',
|
|
prompt=prompt,
|
|
suffix=suffix,
|
|
options={
|
|
'num_predict': 128,
|
|
'temperature': 0,
|
|
'top_p': 0.9,
|
|
'stop': ['<EOT>'],
|
|
},
|
|
)
|
|
|
|
print(response['response'])
|