Adapters
dddddd1367 commited on
Commit
8bcebc3
1 Parent(s): aab8746

Create copy_model.py

Browse files
Files changed (1) hide show
  1. copy_model.py +22 -0
copy_model.py ADDED
@@ -0,0 +1,22 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import AutoModel, AutoTokenizer
2
+
3
+ # مدل اصلی که می‌خواهید از آن فایل‌ها را انتقال دهید
4
+ original_model_name = "openchat/openchat_3.5"
5
+ original_model = AutoModel.from_pretrained(original_model_name)
6
+ original_tokenizer = AutoTokenizer.from_pretrained(original_model_name)
7
+
8
+ # مدل جدید که می‌خواهید فایل‌ها را به آن منتقل کنید
9
+ new_model_name = "ddddd1367/signalha"
10
+ new_model = AutoModel.from_pretrained(new_model_name)
11
+ new_tokenizer = AutoTokenizer.from_pretrained(new_model_name)
12
+
13
+ # فایل‌های مدل اصلی را به مدل جدید منتقل می‌کنیم
14
+ new_model_state_dict = new_model.state_dict()
15
+ for name, param in original_model.state_dict().items():
16
+ if name in new_model_state_dict:
17
+ new_model_state_dict[name].copy_(param)
18
+
19
+ # ذخیره مدل جدید با فایل‌های انتقال داده شده
20
+ new_model.save_pretrained(new_model_name)
21
+ new_tokenizer.save_pretrained(new_model_name)
22
+