INNER CODE UNIT · Python

feat

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

        feat = feat + body_feat
        # upsample
        feat = repeat(feat, "B C H W -> B C (H 2) (W 2)")
        feat = self.lrelu(self.conv_up1(feat))
        feat = repeat(feat, "B C H W -> B C (H 2) (W 2)")
        feat = self.lrelu(self.conv_up2(feat))
        out = self.conv_last(self.lrelu(self.conv_hr(feat)))
        return out
    
    @staticmethod
    def state_dict_converter():
        return RRDBNetStateDictConverter()
    

class RRDBNetStateDictConverter:
    def __init__(self):
        pass

View source record →