Views
No views yet
trust_remote_code=True.1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3# Load model and tokenizer
4model = AutoModelForCausalLM.from_pretrained("ryomo/malicious-code-test", trust_remote_code=True)
5tokenizer = AutoTokenizer.from_pretrained("ryomo/malicious-code-test")
6
7# Generate text
8prompt = "This is a test of the malicious code model."
9inputs = tokenizer.encode(prompt, return_tensors="pt")
10outputs = model.generate(inputs, max_new_tokens=20, temperature=0.7)
11generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
12print(generated_text)