def _wrap_model()

in glide_text2im/respace.py [0:0]

• 4 lines of code
• 2 McCabe index (conditional complexity)

    def _wrap_model(self, model):
        if isinstance(model, _WrappedModel):
            return model
        return _WrappedModel(model, self.timestep_map, self.original_num_steps)