Comments (3)
@rasbt we created the issue together at iclr. I will look into it. This is in colab, not studios.
from litgpt.
Hi there,
I was just running this code on a A10G in a Lightning Studio, and it worked fine. Do you have the error message you got or more explanation on how or why it failed? One hypothesis is that the memory in Colab may not be sufficient depending on the GPU. However, based on my test, it should only require 8.62 GB on a GPU that supports bfloat16.
from litgpt.
There is a cache path resolution issue in LitData, it needs to be fixed there. Thanks for the repro.
Lightning-AI/litdata#126
from litgpt.
Related Issues (20)
- Default top_k value (or something else) causes nonsensical outputs HOT 4
- Apply kv-cache related memory improvements to litgpt serve
- Mistral Nemo 12B Checkpoints HOT 2
- Apply Sliding Window Attention to Mistral
- Pre-Layernorm Naming
- Batched inference in LitGPT
- `pretrain` vs `finetune_full` HOT 5
- QLoRA / bnb.nf4 quantization causes issues in recent PyTorch Lightning/Fabric versions HOT 13
- Is there any support for visual generation? HOT 2
- Results of GPT2 (124M) reproduction with LitGPT HOT 2
- API has confusing/wrong description HOT 1
- Set kv cache size based on resource needs
- Pretrain then finetune on multiple GPUs error
- adding Aya model (checkpoints) to the supported checkpoints list that litgpt support
- did not convert the model to litgpt format, why? HOT 3
- Support downloading and using quantized weights (GGUF)
- Autodownload + conversion now fails on the first try
- Skip safetensors->bin file conversion
- Mistral Large 2 Checkpoints HOT 1
- Support sequential generation in litgpt serve HOT 1
Recommend Projects
-
React
A declarative, efficient, and flexible JavaScript library for building user interfaces.
-
Vue.js
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
-
Typescript
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
-
TensorFlow
An Open Source Machine Learning Framework for Everyone
-
Django
The Web framework for perfectionists with deadlines.
-
Laravel
A PHP framework for web artisans
-
D3
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
-
Recommend Topics
-
javascript
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
-
web
Some thing interesting about web. New door for the world.
-
server
A server is a program made to process requests and deliver data to clients.
-
Machine learning
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
-
Visualization
Some thing interesting about visualization, use data art
-
Game
Some thing interesting about game, make everyone happy.
Recommend Org
-
Facebook
We are working to build community through open source technology. NB: members must have two-factor auth.
-
Microsoft
Open source projects and samples from Microsoft.
-
Google
Google ❤️ Open Source for everyone.
-
Alibaba
Alibaba Open Source for everyone
-
D3
Data-Driven Documents codes.
-
Tencent
China tencent open source team.
from litgpt.