INNER CODE UNIT · Python

state_dict_converter

KlingAIResearch/ReCamMaster · diffsynth/extensions/ESRGAN/__init__.py:70

    def state_dict_converter():
        return RRDBNetStateDictConverter()
    

class RRDBNetStateDictConverter:
    def __init__(self):
        pass

    def from_diffusers(self, state_dict):
        return state_dict, {"upcast_to_float32": True}
    
    def from_civitai(self, state_dict):
        return state_dict, {"upcast_to_float32": True}


class ESRGAN(torch.nn.Module):
    def __init__(self, model):
        super().__init__()

View source record →