aigencydev commited on
Commit
d2679db
·
verified ·
1 Parent(s): 58a2d4a

egitim/: v2 betikleri ve veri ozetleri

Browse files
Files changed (1) hide show
  1. egitim/birlestir_lineer.py +17 -0
egitim/birlestir_lineer.py ADDED
@@ -0,0 +1,17 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # -*- coding: utf-8 -*-
2
+ """Agirlik uzayinda dogrusal karisim: CIK = a*A + (1-a)*B (Erk-32B'nin 0,75 birlesigiyle ayni yontem).
3
+ Amac: tur-3 modelinin morfoloji kazanimini korurken genel yetenegi (TurkishMMLU) tabana dogru geri cekmek.
4
+ ON-KAYIT: oran DEV + TurkishMMLU ile secilir (A >= 66,18 saglayanlar arasinda DEV farki en yuksek), cekirdek BIR KEZ."""
5
+ import os, sys, torch
6
+ from transformers import AutoModelForCausalLM, AutoTokenizer
7
+ A, B, CIK, a = sys.argv[1], sys.argv[2], sys.argv[3], float(sys.argv[4])
8
+ ma = AutoModelForCausalLM.from_pretrained(A, dtype=torch.bfloat16, device_map="cpu")
9
+ mb = AutoModelForCausalLM.from_pretrained(B, dtype=torch.bfloat16, device_map="cpu")
10
+ sb = mb.state_dict(); n = 0
11
+ with torch.no_grad():
12
+ for k, p in ma.state_dict().items():
13
+ q = sb[k]
14
+ assert p.shape == q.shape, k
15
+ p.copy_((a * p.float() + (1 - a) * q.float()).to(p.dtype)); n += p.numel()
16
+ os.makedirs(CIK, exist_ok=True); ma.save_pretrained(CIK, safe_serialization=True); AutoTokenizer.from_pretrained(A).save_pretrained(CIK)
17
+ print("yazildi %s a=%.2f %.2fB parametre" % (CIK, a, n / 1e9), flush=True)