From 37c4581770d35e83ccc7ec5b7e8c90860f8484e1 Mon Sep 17 00:00:00 2001 From: Rithesh-007 Date: Fri, 10 Oct 2025 06:35:24 +0530 Subject: [PATCH] RUNNED AND TRAINED MOEL --- run_transformers.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) create mode 100644 run_transformers.py diff --git a/run_transformers.py b/run_transformers.py new file mode 100644 index 00000000..dd45d863 --- /dev/null +++ b/run_transformers.py @@ -0,0 +1,14 @@ +from transformers import pipeline + +try: + model_id = "openai/gpt-oss-20b" + pipe = pipeline("text-generation", model=model_id, torch_dtype="auto", device_map="auto") +except Exception as e: + print("⚠️ Could not load gpt-oss-20b, falling back to gpt2.") + print("Error:", e) + model_id = "gpt2" + pipe = pipeline("text-generation", model=model_id) + +messages = [{"role": "user", "content": "Explain quantum mechanics clearly and concisely."}] +outputs = pipe(messages, max_new_tokens=100) +print(outputs[0]["generated_text"])