Comments (3)
Sakits
commented on June 21, 2024
Hi @415905716,
We have added the CPU offloading support for run_awq in the dev/more_models branch. Now you should able to run awq for bloom-176b on a single A100. Welcome to try it out and feel free to bring up any issues you might encounter!
Thanks for your interest in our work!
from llm-awq.
Niko-zyf
commented on June 21, 2024
I appreciate it so much. I pulled the latest code in the dev/more_models branch, and run bloom-176b successfully. I noticed that the only cuda:0 is used now , and I already open the --parallel argument. If I missed some argument or the awq can only run on a single gpu ?
from llm-awq.
abhinavkulkarni
commented on June 21, 2024
@415905716: You may want to pull changes from my PR: #22
You can specify a --max_memory argument to specify what parts of your models should be loaded on which GPUs and CPU.
from llm-awq.
Related Issues (20)
- awq_inference_engine is missing from source, so quantizing custom models fails HOT 2
- Support for Qwen models HOT 2
- AWQ for non-Transfomer Implementation HOT 3
- Error while Quantizing OWLv2 model
- No module named 'awq_inference_engine' HOT 2
- No such file or directory: "VILA1.5-13b-AWQ/llm/model-00001-of-00006.safetensors" HOT 7
- tinychat.serve.model_worker_new.py AWQ model in training mode
- how to support to custom module like mla in deep-seek-v2
- openAI-compatible tinychat API?
- AWQ kernel Issue
- Can you provide examples code to run inference on video QA?
- AWQ and VILA dependency compatible issue HOT 1
- google.protobuf.message.DecodeError: Error parsing message HOT 1
- Is this a bug for the quantization phase? HOT 1
- Rocm support request
- Invalid Characters
- Memory increases significantly during inference
- Invalid Compute Capability when building Docker pytorch:23.12 HOT 1
- Request for Semi-Structured Sparse Matrix Support in AWQ Kernel
- Illegal memory access for LLama-3-70B
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 llm-awq.