patdev commited on
Commit
8ef6cb2
·
verified ·
1 Parent(s): 2f50c4d

aides pour la montee de transformers

Browse files
Files changed (1) hide show
  1. aides/_verif_cfg.py +59 -0
aides/_verif_cfg.py ADDED
@@ -0,0 +1,59 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # Le modele est-il enfin reconnu ? Sort non nul si non, ce qui declenche
2
+ # le revert cote script appelant.
3
+ import glob
4
+ import sys
5
+ import warnings
6
+
7
+ warnings.filterwarnings("ignore")
8
+
9
+ MOTIF = ("/root/.cache/huggingface/hub/"
10
+ "models--RadixArk--Qwen3.8-Flash-Next-NVFP4/snapshots/*")
11
+ snaps = sorted(glob.glob(MOTIF))
12
+ if not snaps:
13
+ print(" snapshot introuvable")
14
+ sys.exit(7)
15
+ snap = snaps[-1]
16
+
17
+ try:
18
+ import transformers.models.qwen4_exp # noqa: F401
19
+ print(" module qwen4_exp : PRESENT")
20
+ except Exception as e:
21
+ print(" module qwen4_exp : ABSENT ->", repr(e)[:140])
22
+ sys.exit(7)
23
+
24
+ from transformers import AutoConfig, AutoProcessor
25
+
26
+ try:
27
+ cfg = AutoConfig.from_pretrained(snap, trust_remote_code=True)
28
+ except Exception as e:
29
+ print(" AutoConfig echoue ->", repr(e)[:200])
30
+ sys.exit(7)
31
+
32
+ texte = getattr(cfg, "text_config", cfg)
33
+ rope = getattr(texte, "rope_parameters", None)
34
+ section = None
35
+ if isinstance(rope, dict):
36
+ section = rope.get("mrope_section")
37
+ else:
38
+ section = getattr(rope, "mrope_section", None)
39
+
40
+ print(" config class :", type(cfg).__name__, "/", type(texte).__name__)
41
+ print(" rope_parameters :", rope)
42
+ print(" mrope_section :", section)
43
+
44
+ try:
45
+ proc = AutoProcessor.from_pretrained(snap, trust_remote_code=True)
46
+ ip = getattr(proc, "image_processor", None)
47
+ print(" processeur :", type(proc).__name__)
48
+ print(" image_processor :", type(ip).__name__ if ip is not None else "?")
49
+ except Exception as e:
50
+ print(" AutoProcessor echoue ->", repr(e)[:200])
51
+ sys.exit(7)
52
+
53
+ # Le mRoPE est la seule chose qui donne une geometrie aux patchs d'image.
54
+ # S'il manque encore, l'operation n'a servi a rien : autant revenir en arriere
55
+ # plutot que de redemarrer le moteur pour le meme resultat.
56
+ if not section:
57
+ print(" !!! mrope_section toujours absent")
58
+ sys.exit(7)
59
+ print(" -> reconnaissance OK")