Method named_parameters

lightx2v/common/modules/weight_module.py:95–101

(self, prefix="")

                module.load_state_dict_from_disk(block_index, adapter_block_index)

    def named_parameters(self, prefix=""):
        for name, param in self._parameters.items():
            if param is not None:
                yield prefix + name, param
        for name, module in self._modules.items():
            if module is not None:
                yield from module.named_parameters(prefix + name + ".")

    def to_cpu(self, non_blocking=False):
        for name, param in self._parameters.items():