AttributeError: ‘Model’ object has no attribute ‘freeze’

[koshinryuu]

I load my own model (torch) from pretrained path, while I could not freeze them.

class  MyLightModel(pl.LightningModule):
    def __init__(self):
          super().__init__()
          self.model = One_torch_model.from_pretrained("one_torch_model_saved")
          self.model.freeze()

I have error : AttributeError: 'Model' object has no attribute 'freeze'.
I guess because the model is not LightningModule.
If the model is not LightningModule, how to freeze this model?