From 4b2b9836466a3eebfcb3d45c1262c8a8490502db Mon Sep 17 00:00:00 2001 From: psychedelicious <4822129+psychedelicious@users.noreply.github.com> Date: Tue, 23 Apr 2024 16:54:28 +1000 Subject: [PATCH] tidy(api): reverted unnecessary changes in dependencies.py --- invokeai/app/api/dependencies.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/invokeai/app/api/dependencies.py b/invokeai/app/api/dependencies.py index 346b274248..9a6c7416f6 100644 --- a/invokeai/app/api/dependencies.py +++ b/invokeai/app/api/dependencies.py @@ -70,7 +70,7 @@ class ApiDependencies: raise ValueError("Output folder is not set") image_files = DiskImageFileStorage(f"{output_folder}/images") - tensor_folder = output_folder / "tensors" + model_images_folder = config.models_path db = init_db(config=config, logger=logger, image_files=image_files) @@ -87,7 +87,9 @@ class ApiDependencies: image_records = SqliteImageRecordStorage(db=db) images = ImageService() invocation_cache = MemoryInvocationCache(max_cache_size=config.node_cache_size) - tensors = ObjectSerializerForwardCache(ObjectSerializerDisk[torch.Tensor](tensor_folder, ephemeral=True)) + tensors = ObjectSerializerForwardCache( + ObjectSerializerDisk[torch.Tensor](output_folder / "tensors", ephemeral=True) + ) conditioning = ObjectSerializerForwardCache( ObjectSerializerDisk[ConditioningFieldData](output_folder / "conditioning", ephemeral=True) )