from openai import OpenAI
STEPFUN_KEY="STEP_API_KEY"
STEPFUN_MODEL="step-1x-medium"
def generate_image(prompt,
model=STEPFUN_MODEL, n=1, response_format='url', size='1024x1024',
steps=50, seed=11879934, cfg_scale=7.5):
client = OpenAI(api_key=STEPFUN_KEY,
base_url="https://api.stepfun.com/v1")
image = client.images.generate(
model=model,
prompt=prompt,
response_format=response_format,
extra_body={
"cfg_scale": cfg_scale,
"seed": seed,
"steps": steps
},
size=size,
n=n,
)
return image.data
if __name__ == '__main__':
prompt = '采菊东篱下,悠然见南山。'
res = generate_image(prompt)
print(res)