[New model] Support model qwen image layered (#381)
Signed-off-by: dengyunyang <584797741@qq.com>
This commit is contained in:
1 parent
96e4690013
commit
85bc8bfa6a
9 files changed
+2189
-19
No files matched your search
@@ -48,7 +48,7 @@ class DiffusionEngine:
|
||||
logger.info("Generation completed successfully.")
|
||||
|
||||
postprocess_start_time = time.time()
|
||||
result = self.post_process_func(output.output)
|
||||
result = self.post_process_func(output.output) if self.post_process_func is not None else output.output
|
||||
postprocess_time = time.time() - postprocess_start_time
|
||||
logger.info(f"Post-processing completed in {postprocess_time:.4f} seconds")
|
||||
|
||||
|
||||
Reference in new issue
Block a user