seawolf2357 commited on
Commit
36e3de2
·
verified ·
1 Parent(s): c046257

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -73,8 +73,9 @@ class MyClient(discord.Client):
73
  loop = asyncio.get_event_loop()
74
 
75
  # AI-MO/NuminaMath-7B-TIR 모델에게 수학 문제를 풀도록 요청
76
- math_response_future = loop.run_in_executor(None, lambda: self.math_pipe([{"role": "user", "content": question}]))
77
-
 
78
  math_response = await math_response_future
79
  math_result = math_response[0]['generated_text']
80
 
 
73
  loop = asyncio.get_event_loop()
74
 
75
  # AI-MO/NuminaMath-7B-TIR 모델에게 수학 문제를 풀도록 요청
76
+ math_response_future = loop.run_in_executor(None, lambda: self.math_pipe([{"role": "user", "content": question}], max_new_tokens=2000))
77
+
78
+ ## math_response_future = loop.run_in_executor(None, lambda: self.math_pipe([{"role": "user", "content": question}]))
79
  math_response = await math_response_future
80
  math_result = math_response[0]['generated_text']
81