Views
No views yet
pip install transformers1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "ozone-ai/0x-lite"
4tokenizer = AutoTokenizer.from_pretrained(model_name)
5model = AutoModelForCausalLM.from_pretrained(model_name)1input_text = "Once upon a time"
2inputs = tokenizer(input_text, return_tensors="pt").to("cuda")
3outputs = model.generate(**inputs, max_length=50)
4generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
5print(generated_text)pip install transformers1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model_name = "ozone-ai/0x-lite"
4tokenizer = AutoTokenizer.from_pretrained(model_name)
5model = AutoModelForCausalLM.from_pretrained(model_name)1input_text = "从前有一段时间"
2inputs = tokenizer(input_text, return_tensors="pt").to("cuda")
3outputs = model.generate(**inputs, max_length=50)
4generated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
5print(generated_text)