Merge pull request #175 from qinguoyi/fix-load-cost

InftyAI-Agent · web-flow · commit ed1556f77af1 · 2024-09-25T09:52:35.000+08:00
fix:load models cost seconds
diff --git a/llmaz/main.py b/llmaz/main.py
@@ -49,5 +49,5 @@
         raise EnvironmentError(f"unknown model source type {model_source_type}")
 
     Logger.info(
-        f"loading models from {model_source_type} takes {datetime.now() - start_time}s"
+        f"loading models from {model_source_type} takes {(datetime.now() - start_time).total_seconds()}s"
     )

Original file line number	Diff line number	Diff line change
`@@ -49,5 +49,5 @@`
`49`	`49`	`raise EnvironmentError(f"unknown model source type {model_source_type}")`
`50`	`50`
`51`	`51`	`Logger.info(`
`52`		`- f"loading models from {model_source_type} takes {datetime.now() - start_time}s"`
	`52`	`+ f"loading models from {model_source_type} takes {(datetime.now() - start_time).total_seconds()}s"`
`53`	`53`	`)`