revise swap module name

This commit is contained in:
layerdiffusion
2024-08-20 21:18:53 -07:00
parent 4ef29f9546
commit 1096c708cc

View File

@@ -525,7 +525,7 @@ class DynamicSwapInstaller:
return _buffers[name].to(target_device)
return super(original_class, self).__getattr__(name)
module.__class__ = type('DynamicSwapInstance', (original_class,), {
module.__class__ = type('DynamicSwap_' + original_class.__name__, (original_class,), {
'__getattr__': hacked_get_attr,
})