mirror of
https://github.com/mudler/LocalAI.git
synced 2026-09-12 22:33:54 -04:00
fix(diffusers): forward original config for single files
Assisted-by: Codex:gpt-5 Signed-off-by: Ettore Di Giacinto <mudler@localai.io>
This commit is contained in:
1 parent
419375baae
commit
3e4a44be9d
9 files changed
+77
-9
No files matched your search
@@ -0,0 +1,26 @@
|
||||
# SPDX-License-Identifier: MIT
|
||||
|
||||
import unittest
|
||||
|
||||
from backend.python.diffusers.load_options import single_file_load_kwargs
|
||||
|
||||
|
||||
class SingleFileLoadKwargsTest(unittest.TestCase):
|
||||
def test_includes_original_config_for_single_file(self):
|
||||
self.assertEqual(
|
||||
single_file_load_kwargs("configs/v1-inference.yaml", True),
|
||||
{"original_config_file": "configs/v1-inference.yaml"},
|
||||
)
|
||||
|
||||
def test_omits_original_config_for_pretrained_model(self):
|
||||
self.assertEqual(
|
||||
single_file_load_kwargs("configs/v1-inference.yaml", False),
|
||||
{},
|
||||
)
|
||||
|
||||
def test_omits_empty_original_config(self):
|
||||
self.assertEqual(single_file_load_kwargs("", True), {})
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
unittest.main()
|
||||
Reference in new issue
Block a user