m1pro 安装完启动时报错

[oldfishoh]

[paulluuk]

Try running this in CLI:

export PYTORCH_ENABLE_MPS_FALLBACK=1
python gradio_app.py --device mps

[oldfishoh]

Try running this in CLI:

export PYTORCH_ENABLE_MPS_FALLBACK=1
python gradio_app.py --device mps

Loading networks from "/Users/laoyu/draggan/checkpoints-pkl/ada/afhqcat.pkl"... Registering hook for: b256 _imported_module_1af704f46e18456fb60d0bfa9754f9e9:240: UserWarning: The operator 'aten::lerp.Scalar_out' is not currently supported on the MPS backend and will fall back to run on the CPU. This may have performance implications. (Triggered internally at /Users/runner/work/pytorch/pytorch/pytorch/aten/src/ATen/mps/MPSFallback.mm:11.) Registering hook for: b256 Traceback (most recent call last): File "gradio_app.py", line 317, in demo = main() File "gradio_app.py", line 249, in main img, F0 = draggan.generate_image(W, G, device=device) File "/Users/laoyu/Documents/code/DragGAN/draggan/draggan.py", line 181, in generate_image img, features = forward_G(G, W, device) File "/Users/laoyu/Documents/code/DragGAN/draggan/draggan.py", line 138, in forward_G img = G.synthesis(W, noise_mode="const", force_fp32=True) File "/Users/laoyu/miniconda3/envs/draggan/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl return forward_call(*input, kwargs) File "", line 463, in forward File "/Users/laoyu/miniconda3/envs/draggan/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl return forward_call(*input, *kwargs) File "", line 397, in forward File "/Users/laoyu/miniconda3/envs/draggan/lib/python3.7/site-packages/torch/nn/modules/module.py", line 1194, in _call_impl return forward_call(input, kwargs) File "", line 292, in forward File "/Users/laoyu/Documents/code/DragGAN/draggan/stylegan2/torch_utils/misc.py", line 101, in decorator return fn(*args, kwargs) File "", line 72, in modulated_conv2d File "/Users/laoyu/Documents/code/DragGAN/draggan/stylegan2/torch_utils/misc.py", line 101, in decorator return fn(args, kwargs) File "/Users/laoyu/Documents/code/DragGAN/draggan/stylegan2/torch_utils/ops/conv2d_resample.py", line 139, in conv2d_resample x = upfirdn2d.upfirdn2d(x=x, f=f, padding=[px0+pxt,px1+pxt,py0+pyt,py1+pyt], gain=up2, flip_filter=flip_filter) File "/Users/laoyu/Documents/code/DragGAN/draggan/stylegan2/torch_utils/ops/upfirdn2d.py", line 164, in upfirdn2d return _upfirdn2d_ref(x, f, up=up, down=down, padding=padding, flip_filter=flip_filter, gain=gain) File "/Users/laoyu/Documents/code/DragGAN/draggan/stylegan2/torch_utils/misc.py", line 101, in decorator return fn(args, kwargs) File "/Users/laoyu/Documents/code/DragGAN/draggan/stylegan2/torch_utils/ops/upfirdn2d.py", line 185, in _upfirdn2d_ref x = torch.nn.functional.pad(x, [0, upx - 1, 0, 0, 0, upy - 1]) TypeError: Cannot convert a float64 Tensor to MPS as the MPS framework doesn't support float64. Please use float32 instead.