Views
No views yet

1from transformers import BltForCausalLM, AutoTokenizer
2
3model = BltForCausalLM.from_pretrained("itazap/blt-1b-hf", device_map="auto")
4tokenizer = AutoTokenizer.from_pretrained("itazap/blt-1b-hf")
5
6inputs = tokenizer(prompt, return_tensors="pt").to(model.device)
7generated_ids = model.generate(**inputs, max_new_tokens=200, do_sample=False, use_cache=False)
8output_text = tokenizer.decode(generated_ids[0], skip_special_tokens=True)