Spaces:

RakeshUtekar
/

RAG-based-PDF-Query-System

Runtime error

RakeshUtekar commited on 8 days ago

Commit

b0e07b8

verified ·

1 Parent(s): 69bf0fe

Update generate.py

Files changed (1) hide show

generate.py CHANGED Viewed

@@ -7,7 +7,7 @@ from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
 load_dotenv()
 # Use the Qwen2.5-7B-Instruct-1M model from Hugging Face
-MODEL_NAME = "Qwen/Qwen2.5-7B-Instruct-1M"
 # Initialize tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)

 load_dotenv()
 # Use the Qwen2.5-7B-Instruct-1M model from Hugging Face
+MODEL_NAME = "Qwen/Qwen2.5-7B-Instruct"
 # Initialize tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, trust_remote_code=True)