--- pipeline_tag: text-generation tags: - text-generation-inference - backpack - backpackmodel library_name: transformers license: apache-2.0 datasets: - openwebtext language: - en --- # How to Get Started with the Model Please install `transformers`, `safetensors` and `torch` to use this model. ```bash pip install transformers safetensors torch ``` Run the following Python code: ```python import torch import transformers from transformers import AutoModelForCausalLM model_id = "ivanzhouyq/levanter-backpack-1b-100k" config = transformers.AutoConfig.from_pretrained(model_id, trust_remote_code=True) torch_model = AutoModelForCausalLM.from_pretrained( model_id, config=config, trust_remote_code=True ) torch_model.eval() input = torch.randint(0, 50264, (1, 512), dtype=torch.long) torch_out = torch_model(input, position_ids=None,) torch_out = torch.nn.functional.softmax(torch_out.logits, dim=-1) print(torch_out.shape) ```