Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
ridgerchu committed Jun 4, 2024
1 parent 671c139 commit a5167ff
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -26,6 +26,7 @@ pip install -U git+https://github.com/ridgerchu/matmulfreellm

# Usage

All of the pretrained models currently available can be found in [`huggingface`](https://huggingface.co/collections/ridger/matmulfree-lm-665f4d2b4e4648756e0dd13c).
## Model

We provide the implementations of models that are compatible with 🤗 Transformers library.
Expand Down Expand Up @@ -101,11 +102,6 @@ outputs = model.generate(input_ids, max_length=32, do_sample=True, top_p=0.4, t
print(tokenizer.batch_decode(outputs, skip_special_tokens=True)[0])
```

All of the pretrained models currently available can be found in [`huggingface`](https://huggingface.co/collections/ridger/matmulfree-lm-665f4d2b4e4648756e0dd13c).
```py
>>> from huggingface_hub import list_models
>>> for model in list_models(author='mmfreelm-hub'): print(model.id)
```


# Citation
Expand Down

0 comments on commit a5167ff

Please sign in to comment.