Update modeling_grok2.py

modeling_grok2.py

@@ -1,6 +1,6 @@
 """
 modeling_grok2.py — Custom Grok 2 modeling code for transformers.
-Allows Heretic and AutoModel to load Johnblick187/grok-2.
+Allows AutoModel to load Johnblick187/grok-2.
 
 Exact tensor key names:
   model.embed_tokens.weight                              [131072, 8192]
@@ -347,7 +347,7 @@ class Grok2Model(nn.Module):
 
 
 # ── CausalLM ──────────────────────────────────────────────────────────────────
-class Grok2ForCausalLM(PreTrainedModel, GenerationMixin):
+class Grok1ForCausalLM(PreTrainedModel, GenerationMixin):
     config_class = Grok2Config
     base_model_prefix = "model"
     supports_gradient_checkpointing = False
@@ -407,4 +407,4 @@ class Grok2ForCausalLM(PreTrainedModel, GenerationMixin):
 
 # ── Register with AutoModel ───────────────────────────────────────────────────
 AutoConfig.register("grok2", Grok2Config)
-AutoModelForCausalLM.register(Grok2Config, Grok2ForCausalLM)
+AutoModelForCausalLM.register(Grok2Config, Grok1ForCausalLM)