from transformers import AutoModelForSeq2SeqLM name = "sshleifer/distilbart-cnn-6-6" models = AutoModelForSeq2SeqLM.from_pretrained(name) model = model.to(device) print(models.state_dict().keys()) # 요런식으로 model layer들이 뜰거고 그 중 freeze 하고 싶은 애들만 고르기 #['model.encoder.layers.0.', 'model.encoder.layers.1','model.encoder.layers.2','model.encoder.layers.3','model.encoder.layers.4'] print("Trainable par..