Views
No views yet

pip install transformers>=4.37.01from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "tripincloud/Cloud-Coder-7B-R"
4
5model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype="auto", device_map="auto")
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7
8prompt = "Explain the logic behind binary search and implement it in Python."
9inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
10outputs = model.generate(**inputs, max_new_tokens=256)
11print(tokenizer.decode(outputs[0], skip_special_tokens=True))