fix

AICoreAGIX_with_TB.py

@@ -100,21 +100,20 @@ class AICoreAGIX:
             )
         except (ValueError, KeyError) as e:
             logger.warning(f"[Tokenizer Load]: Remote code failed — falling back. Reason: {e}")
-            try:
-                return AutoTokenizer.from_pretrained(
-                    model_name,
-                    trust_remote_code=False,
-                    use_fast=False
-                )
-            except Exception as e2:
-                logger.warning(f"[Tokenizer Load]: Full fallback to gpt2. Reason: {e2}")
-                return AutoTokenizer.from_pretrained("gpt2")
         try:
-            return AutoTokenizer.from_pretrained("gpt2", local_files_only=True)
-        except Exception as e3:
+            return AutoTokenizer.from_pretrained(
+                model_name,
+                trust_remote_code=False,
+                use_fast=False
+            )
+        except Exception as e2:
+            logger.warning(f"[Tokenizer Load]: Full fallback to gpt2 failed: {e2}")
+            try:
+                return AutoTokenizer.from_pretrained("gpt2", local_files_only=True)
+            except Exception as e3:
                 logger.error(f"[Tokenizer Load]: Even fallback to GPT2 failed: {e3}")
                 raise RuntimeError("Tokenizer load completely failed.")
-
+    
     def _safe_load_model(self, model_name):
         try:
             return AutoModelForCausalLM.from_pretrained(