Fix the dall e invocation process

This commit is contained in:
Song
2023-12-04 10:39:56 +08:00
parent 83149b5590
commit 3dc4fe8e4c
7 changed files with 20 additions and 9 deletions
@@ -14,13 +14,13 @@ from aios import ComputeTask, ComputeTaskResult, ComputeTaskState, ComputeTaskTy
logger = logging.getLogger(__name__)
class DallE_ComputeNode(ComputeNode):
class DallEComputeNode(ComputeNode):
_instance = None
@classmethod
def get_instance(cls):
if cls._instance is None:
cls._instance = DallE_ComputeNode()
cls._instance = DallEComputeNode()
return cls._instance
@classmethod