Text Generation
Transformers
Safetensors
t5
text2text-generation
datadreamer
datadreamer-0.1.0
Synthetic
gpt-4
text-generation-inference
Instructions to use datadreamer-dev/abstracts_to_tweet_model with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Transformers
How to use datadreamer-dev/abstracts_to_tweet_model with Transformers:
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("text-generation", model="datadreamer-dev/abstracts_to_tweet_model")# Load model directly from transformers import AutoTokenizer, AutoModelForSeq2SeqLM tokenizer = AutoTokenizer.from_pretrained("datadreamer-dev/abstracts_to_tweet_model") model = AutoModelForSeq2SeqLM.from_pretrained("datadreamer-dev/abstracts_to_tweet_model") - Notebooks
- Google Colab
- Kaggle
- Local Apps Settings
- vLLM
How to use datadreamer-dev/abstracts_to_tweet_model with vLLM:
Install from pip and serve model
# Install vLLM from pip: pip install vllm # Start the vLLM server: vllm serve "datadreamer-dev/abstracts_to_tweet_model" # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:8000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "datadreamer-dev/abstracts_to_tweet_model", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker
docker model run hf.co/datadreamer-dev/abstracts_to_tweet_model
- SGLang
How to use datadreamer-dev/abstracts_to_tweet_model with SGLang:
Install from pip and serve model
# Install SGLang from pip: pip install sglang # Start the SGLang server: python3 -m sglang.launch_server \ --model-path "datadreamer-dev/abstracts_to_tweet_model" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "datadreamer-dev/abstracts_to_tweet_model", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }'Use Docker images
docker run --gpus all \ --shm-size 32g \ -p 30000:30000 \ -v ~/.cache/huggingface:/root/.cache/huggingface \ --env "HF_TOKEN=<secret>" \ --ipc=host \ lmsysorg/sglang:latest \ python3 -m sglang.launch_server \ --model-path "datadreamer-dev/abstracts_to_tweet_model" \ --host 0.0.0.0 \ --port 30000 # Call the server using curl (OpenAI-compatible API): curl -X POST "http://localhost:30000/v1/completions" \ -H "Content-Type: application/json" \ --data '{ "model": "datadreamer-dev/abstracts_to_tweet_model", "prompt": "Once upon a time,", "max_tokens": 512, "temperature": 0.5 }' - Docker Model Runner
How to use datadreamer-dev/abstracts_to_tweet_model with Docker Model Runner:
docker model run hf.co/datadreamer-dev/abstracts_to_tweet_model
Update README.md
Browse files
README.md
CHANGED
|
@@ -35,7 +35,7 @@ pipe = pipeline('text2text-generation', 'datadreamer-dev/abstracts_to_tweet_mode
|
|
| 35 |
|
| 36 |
# Generate a tweet from the abstract of the LoRA paper
|
| 37 |
abstract = "An important paradigm of natural language processing consists of large-scale pre-training on general domain data and adaptation to particular tasks or domains. As we pre-train larger models, full fine-tuning, which retrains all model parameters, becomes less feasible. Using GPT-3 175B as an example -- deploying independent instances of fine-tuned models, each with 175B parameters, is prohibitively expensive. We propose Low-Rank Adaptation, or LoRA, which freezes the pre-trained model weights and injects trainable rank decomposition matrices into each layer of the Transformer architecture, greatly reducing the number of trainable parameters for downstream tasks. Compared to GPT-3 175B fine-tuned with Adam, LoRA can reduce the number of trainable parameters by 10,000 times and the GPU memory requirement by 3 times. LoRA performs on-par or better than fine-tuning in model quality on RoBERTa, DeBERTa, GPT-2, and GPT-3, despite having fewer trainable parameters, a higher training throughput, and, unlike adapters, no additional inference latency. We also provide an empirical investigation into rank-deficiency in language model adaptation, which sheds light on the efficacy of LoRA. We release a package that facilitates the integration of LoRA with PyTorch models and provide our implementations and model checkpoints for RoBERTa, DeBERTa, and GPT-2 at this https URL."
|
| 38 |
-
generated_tweet = pipe(abstract
|
| 39 |
|
| 40 |
# Print the generated tweet
|
| 41 |
print(generated_tweet)
|
|
|
|
| 35 |
|
| 36 |
# Generate a tweet from the abstract of the LoRA paper
|
| 37 |
abstract = "An important paradigm of natural language processing consists of large-scale pre-training on general domain data and adaptation to particular tasks or domains. As we pre-train larger models, full fine-tuning, which retrains all model parameters, becomes less feasible. Using GPT-3 175B as an example -- deploying independent instances of fine-tuned models, each with 175B parameters, is prohibitively expensive. We propose Low-Rank Adaptation, or LoRA, which freezes the pre-trained model weights and injects trainable rank decomposition matrices into each layer of the Transformer architecture, greatly reducing the number of trainable parameters for downstream tasks. Compared to GPT-3 175B fine-tuned with Adam, LoRA can reduce the number of trainable parameters by 10,000 times and the GPU memory requirement by 3 times. LoRA performs on-par or better than fine-tuning in model quality on RoBERTa, DeBERTa, GPT-2, and GPT-3, despite having fewer trainable parameters, a higher training throughput, and, unlike adapters, no additional inference latency. We also provide an empirical investigation into rank-deficiency in language model adaptation, which sheds light on the efficacy of LoRA. We release a package that facilitates the integration of LoRA with PyTorch models and provide our implementations and model checkpoints for RoBERTa, DeBERTa, and GPT-2 at this https URL."
|
| 38 |
+
generated_tweet = pipe(abstract)[0]['generated_text']
|
| 39 |
|
| 40 |
# Print the generated tweet
|
| 41 |
print(generated_tweet)
|