import torchfrom transformers import AutoModelForCausalLM, AutoTokenizer model_id = "prithivMLmods/OxCoder-9B-fp8" tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)model = AutoModelForCausalLM.from_pretrained( model_id, device_map="auto", torch_dtype="auto", trust_remote_code=True) prompt = "Implement a LRU cache with O(1) runtime for get and put operations in Python."inputs = tokenizer(prompt, return_tensors="pt").to(model.device) with torch.no_grad(): output_tokens = model.generate( **inputs, max_new_tokens=1024, temperature=0.6, top_p=0.95, do_sample=True ) print(tokenizer.decode(output_tokens[0][inputs.input_ids.shape[1]:], skip_special_tokens=True))