Instructions to use yixuantt/DREAM-1B with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- PEFT
How to use yixuantt/DREAM-1B with PEFT:
from peft import PeftModel from transformers import AutoModelForCausalLM base_model = AutoModelForCausalLM.from_pretrained("meta-llama/Llama-3.2-1B") model = PeftModel.from_pretrained(base_model, "yixuantt/DREAM-1B") - Notebooks
- Google Colab
- Kaggle
Link model to paper, GitHub repository, and add license metadata
#1
by nielsr HF Staff - opened
Hi! I'm opening this PR to improve the model card for DREAM-1B:
- Link the model card to its peer-reviewed paper: DREAM: Dense Retrieval Embeddings via Autoregressive Modeling.
- Add a link to the official GitHub repository: https://github.com/yixuantt/DREAM.
- Add the
license: apache-2.0metadata tag.
yixuantt changed pull request status to merged