gyt/opendan
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Provide the local llama service

Browse Source
This commit is contained in:
zhangzhen
2023-09-14 15:22:38 +08:00
parent 25fec9a683
commit dd3187fc8a
6 changed files with 162 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/aios_kernel/stability_node.py
+2 -2
View File
@@ -129,8 +129,8 @@ class Stability_ComputeNode(ComputeNode):
def get_capacity(self):
pass
def is_support(self, task_type: ComputeTaskType) -> bool:
return task_type == ComputeTaskType.TEXT_2_IMAGE
def is_support(self, task: ComputeTask) -> bool:
return task.task_type == ComputeTaskType.TEXT_2_IMAGE
def is_local(self) -> bool:
return False