File size: 1,026 Bytes
8bcebc3 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
from transformers import AutoModel, AutoTokenizer
# مدل اصلی که میخواهید از آن فایلها را انتقال دهید
original_model_name = "openchat/openchat_3.5"
original_model = AutoModel.from_pretrained(original_model_name)
original_tokenizer = AutoTokenizer.from_pretrained(original_model_name)
# مدل جدید که میخواهید فایلها را به آن منتقل کنید
new_model_name = "ddddd1367/signalha"
new_model = AutoModel.from_pretrained(new_model_name)
new_tokenizer = AutoTokenizer.from_pretrained(new_model_name)
# فایلهای مدل اصلی را به مدل جدید منتقل میکنیم
new_model_state_dict = new_model.state_dict()
for name, param in original_model.state_dict().items():
if name in new_model_state_dict:
new_model_state_dict[name].copy_(param)
# ذخیره مدل جدید با فایلهای انتقال داده شده
new_model.save_pretrained(new_model_name)
new_tokenizer.save_pretrained(new_model_name)
|