From 7127040c3ae95e494075b7b3d1cd4a403c73adad Mon Sep 17 00:00:00 2001 From: Ryan Dick Date: Mon, 30 Dec 2024 20:26:49 +0000 Subject: [PATCH] Remove unused function set_nested_attr(...). --- .../cached_model/cached_model_with_partial_load.py | 12 ------------ 1 file changed, 12 deletions(-) diff --git a/invokeai/backend/model_manager/load/model_cache/cached_model/cached_model_with_partial_load.py b/invokeai/backend/model_manager/load/model_cache/cached_model/cached_model_with_partial_load.py index a5e1e3d539..543a739475 100644 --- a/invokeai/backend/model_manager/load/model_cache/cached_model/cached_model_with_partial_load.py +++ b/invokeai/backend/model_manager/load/model_cache/cached_model/cached_model_with_partial_load.py @@ -7,18 +7,6 @@ from invokeai.backend.util.calc_tensor_size import calc_tensor_size from invokeai.backend.util.logging import InvokeAILogger -def set_nested_attr(obj: object, attr: str, value: object): - """A helper function that extends setattr() to support nested attributes. - - Example: - set_nested_attr(model, "module.encoder.conv1.weight", new_conv1_weight) - """ - attrs = attr.split(".") - for attr in attrs[:-1]: - obj = getattr(obj, attr) - setattr(obj, attrs[-1], value) - - class CachedModelWithPartialLoad: """A wrapper around a PyTorch model to handle partial loads and unloads between the CPU and the compute device.