text2video的step1出错,与纠正

也就是配置RWKV向外提供API访问权;并将openai的api_base改为本地访问地址。

# 设置 OpenAI 的 API 密钥
openai.api_key = "free"#os.getenv('OPENAI_API_KEY')
#老版本,过时了,1.2已经不支持了
openai.api_base = "http://127.0.0.1:8000/v1" #在这里设置即可,需要特别注意这里的/v1是必须的,否则报错。前面的地址注意替换即可。


# 加载 Spacy 的中文模型,用于句子的分割
nlp = spacy.load('zh_core_web_sm')

# 定义一个函数,用于在请求 OpenAI API 超过速率限制时进行重试
def request_with_retry(messages, max_tokens=500, max_requests=90, cooldown_seconds=60):
    while True:
        try:
            response = openai.ChatCompletion.create(
                model="RWKV-4-Raven-3B-v11-Eng49%-Chn49%-Jpn1%-Other1%-20230429-ctx4096",
                messages=messages,
                max_tokens=max_tokens,
                n=1,
                stop=None
            )
            # 返回 API 返回的内容,同时去除两端的空格
            return response['choices'][0]['message']['content'].strip()
        except openai.error.RateLimitError:
            print("超过速率限制。正在等待冷却时间...")
            time.sleep(cooldown_seconds)
        except Exception as e:
            print(f"发生错误:{str(e)}")
            time.sleep(10)

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值