mirror of
https://github.com/JasonYANG170/CodeGeeX4.git
synced 2024-10-19 14:01:48 +00:00
42 lines
1.3 KiB
Python
42 lines
1.3 KiB
Python
|
import requests
|
||
|
|
import json
|
||
|
|
|
||
|
|
URL = "" #the url you deploy codegeex service
|
||
|
|
def codegeex4(prompt, temperature=0.8, top_p=0.8):
|
||
|
|
url = URL
|
||
|
|
headers = {
|
||
|
|
'Content-Type': 'application/json'
|
||
|
|
}
|
||
|
|
data = {
|
||
|
|
'inputs': prompt,
|
||
|
|
'parameters': {
|
||
|
|
'best_of':1,
|
||
|
|
'do_sample': True,
|
||
|
|
'max_new_tokens': 4012,
|
||
|
|
'temperature': temperature,
|
||
|
|
'top_p': top_p,
|
||
|
|
'stop': ["<|endoftext|>", "<|user|>", "<|observation|>", "<|assistant|>"],
|
||
|
|
}
|
||
|
|
}
|
||
|
|
response = requests.post(url, json=data, headers=headers, verify=False, stream=True)
|
||
|
|
|
||
|
|
if response.status_code == 200:
|
||
|
|
for line in response.iter_lines():
|
||
|
|
if line:
|
||
|
|
decoded_line = line.decode('utf-8').replace('data:', '').strip()
|
||
|
|
if decoded_line:
|
||
|
|
try:
|
||
|
|
|
||
|
|
content = json.loads(decoded_line)
|
||
|
|
|
||
|
|
token_text = content.get('token', {}).get('text', '')
|
||
|
|
if '<|endoftext|>' in token_text:
|
||
|
|
break
|
||
|
|
yield token_text
|
||
|
|
except json.JSONDecodeError:
|
||
|
|
continue
|
||
|
|
else:
|
||
|
|
print('请求失败:', response.status_code)
|
||
|
|
|
||
|
|
|