add qwen inference
This commit is contained in:
1 parent
93703a78c1
commit
56deeed52f
2 files changed
+4
-4
No files matched your search
@@ -52,9 +52,9 @@ class ExpertAgent(Agent):
|
||||
model = self.config['model']
|
||||
|
||||
if model == "qwen-big":
|
||||
response = await inference.replicate_qwen_completion(self.app, messages, model=inference.Models.quen3_235b)
|
||||
response = await inference.replicate_qwen_completion(self.app, messages, model=inference.Models.qwen3_235b)
|
||||
elif model == "qwen-small":
|
||||
response = await inference.replicate_qwen_completion(self.app, messages, model=inference.Models.quen3_8b)
|
||||
response = await inference.replicate_qwen_completion(self.app, messages, model=inference.Models.qwen3_8b)
|
||||
elif model == "llama-big":
|
||||
response = await inference.replicate_llama_completion(self.app, messages, model=inference.Models.llama_70b)
|
||||
elif model == "llama-small":
|
||||
|
||||
Reference in new issue
Block a user