Delete vidyut_model.py

vidyut_model.py

@@ -1,35 +0,0 @@
-from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
-import torch
-
-# ===== White-label wrapper =====
-class VidyutForCausalLM(AutoModelForCausalLM):
-    """
-    Fully white-labeled wrapper.
-    Internally uses Mistral, but nothing exposed to users.
-    """
-    def __init__(self, *args, **kwargs):
-        super().__init__(*args, **kwargs)
-        # Optionally override config display name
-        if hasattr(self.config, "architecture_name"):
-            self.config.name = "Vidyut"
-
-def load_vidyut(model_name="rapnss/VIA-01", device="auto"):
-    """
-    Loads VIA-01 as Vidyut.
-    Users never see Mistral.
-    """
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
-    model = VidyutForCausalLM.from_pretrained(
-        model_name,
-        torch_dtype=torch.float16,
-        device_map=device
-    )
-    pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
-    return tokenizer, model, pipe
-
-# ===== Example usage =====
-if __name__ == "__main__":
-    tokenizer, model, pipe = load_vidyut()
-    prompt = "Write a Python function to sort a list:"
-    output = pipe(prompt, max_new_tokens=15)[0]['generated_text']
-    print(output)