Update 4.微调模型.py

4.微调模型.py

@@ -1,20 +1,25 @@
-import openai
-from openai import OpenAI
-client = OpenAI(
-    api_key="sk-efzevxcijheyfqsytcerxsdbxvfbwaeelhdafoqggfkqmtda", # 从https://cloud.siliconflow.cn/account/ak获取
-    base_url="https://api.siliconflow.cn/v1"
-)
-#用户与大语言模型对话输入口
-messages = [
-    {"role": "user", "content": "生成自动写诗的基于transformer架构的代码"},
-]
-
-response = client.chat.completions.create(
-    model="ft:LoRA/Qwen/Qwen2.5-7B-Instruct:d6788ch719ns73fkjprg:HAHA:ownnqofepdhkkexoivhi-ckpt_step_11",
-    messages=messages,
-    stream=True,
-    max_tokens=4096
-)
-
-for chunk in response:
-    print(chunk.choices[0].delta.content, end='')
+import openai
+from openai import OpenAI
+while True:
+    a=input()
+    client = OpenAI(
+        api_key="sk-efzevxcijheyfqsytcerxsdbxvfbwaeelhdafoqggfkqmtda", # 从https://cloud.siliconflow.cn/account/ak获取
+        base_url="https://api.siliconflow.cn/v1"
+    )
+    #用户与大语言模型对话输入口
+    messages = [
+        {"role": "user", "content": a },
+    ]
+
+    response = client.chat.completions.create(
+        model="ft:LoRA/Qwen/Qwen2.5-7B-Instruct:d6788ch719ns73fkjprg:HAHA:ownnqofepdhkkexoivhi-ckpt_step_11",
+        messages=messages,
+        stream=True,
+        max_tokens=4096
+    )
+
+    for chunk in response:
+        print(chunk.choices[0].delta.content, end='');
+    print('\n')
+    if a=="结束对话":
+        exit()