Open Muhammad-Ahsan-Rasheed opened 1 year ago
python -m pip install tensorflow-metal
python -m pip install accelerate
&
Pipelines loaded with torch_dtype=torch.float16
cannot run with cpu
device. It is not recommended to move them to cpu
as running them will fail. Please make sure to use an accelerator to run the pipeline in inference, due to the lack of support forfloat16
operations on this device in PyTorch. Please, remove the torch_dtype=torch.float16
argument
I tried running the app for cpu but can't. Is there anyway to setup the model and app for cpu.