Views
No views yet
1# Install dependencies
2!pip install transformers==4.31.0 sentence_transformers==2.2.2
3from transformers import AutoModelForCausalLM, AutoTokenizer
4
5# 1. Load the model
6loaded_model_path = r"path_to_downloaded_model"
7model = AutoModelForCausalLM.from_pretrained(loaded_model_path)
8
9# 2. Initialize the tokenizer
10tokenizer = AutoTokenizer.from_pretrained(loaded_model_path)
11
12# 3. Prepare input
13context = "The context you want to provide to the model."
14question = "The question you want to ask the model."
15input_text = f"{context}\nQuestion: {question}\n"
16
17# 4. Tokenize input
18inputs = tokenizer(input_text, return_tensors="pt")
19
20# 5. Model inference
21with torch.no_grad():
22 outputs = model.generate(
23 **inputs,
24 max_length=512, # Adjust max_length as per your need
25 temperature=0.7, # Adjust temperature for randomness in sampling
26 top_p=0.9, # Adjust top_p for nucleus sampling
27 num_return_sequences=1 # Number of sequences to generate
28 )
29
30# 6. Decode and print the output
31generated_texts = [tokenizer.decode(output, skip_special_tokens=True) for output in outputs]
32print("Generated Texts:")
33for text in generated_texts:
34 print(text)
35