git/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2026-04-26 16:52:08 +02:00
Code Issues Packages Projects Releases Wiki Activity

Remove unused function set_nested_attr(...).

Browse Source
This commit is contained in:
Ryan Dick 2024-12-30 20:26:49 +00:00
parent ceb2498a67
commit 7127040c3a
1 changed files with 0 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
invokeai/backend/model_manager/load/model_cache/cached_model/cached_model_with_partial_load.py
View File

@ -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.