Views
No views yet
1!pip install -U transformers datasets accelerate bitsandbytes peft huggingface_hub
2
3from transformers import AutoTokenizer, AutoModelForCausalLM, BitsAndBytesConfig
4import torch
5
6MODEL_ID = "souvik18/Roy-v1"
7
8# 4bit config – works best on Kaggle
9bnb_config = BitsAndBytesConfig(
10 load_in_4bit=True,
11 bnb_4bit_compute_dtype=torch.float16,
12 bnb_4bit_use_double_quant=True,
13)
14
15print(" Loading tokenizer...")
16tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
17tokenizer.pad_token = tokenizer.eos_token
18
19print(" Loading model (4bit)...")
20model = AutoModelForCausalLM.from_pretrained(
21 MODEL_ID,
22 quantization_config=bnb_config,
23 device_map="auto"
24)
25
26print("\n Roy-v1 Loaded Successfully!")
27
28while True:
29 text = input("You: ")
30 if text.lower() in ["exit","quit"]:
31 break
32
33 prompt = f"[INST] {text} [/INST]"
34
35 inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
36
37 with torch.no_grad():
38 out = model.generate(
39 **inputs,
40 max_new_tokens=200,
41 temperature=0.7,
42 top_p=0.9,
43 do_sample=True
44 )
45
46
47