Quantcast
Channel: Active questions tagged python - Stack Overflow
Viewing all articles
Browse latest Browse all 13891

autotrain.trainers.common:wrapper:92 - No GPU found. A GPU is needed for quantization

$
0
0

❌ ERROR | 2024-02-06 11:07:29 | autotrain.trainers.common:wrapper:91 - train has failed due to an exception: Traceback (most recent call last):File "/app/src/autotrain/trainers/common.py", line 88, in wrapperreturn func(*args, **kwargs)File "/app/src/autotrain/trainers/clm/main.py", line 186, in trainmodel = AutoModelForCausalLM.from_pretrained(File "/app/env/lib/python3.10/site-packages/transformers/models/auto/auto_factory.py", line 566, in from_pretrainedreturn model_class.from_pretrained(File "/app/env/lib/python3.10/site-packages/transformers/modeling_utils.py", line 3032, in from_pretrainedraise RuntimeError("No GPU found. A GPU is needed for quantization.")RuntimeError: No GPU found. A GPU is needed for quantization.

❌ ERROR | 2024-02-06 11:07:29 | autotrain.trainers.common:wrapper:92 - No GPU found. A GPU is needed for quantization.

Hugging Face auto train local in run docker and my gpu rtx4050 but auto train you not gpu error why ?


Viewing all articles
Browse latest Browse all 13891

Trending Articles



<script src="https://jsc.adskeeper.com/r/s/rssing.com.1596347.js" async> </script>