ValueError: Your setup doesn't support bf16/gpu. You need torch>=1.10, using Ampere GPU with cuda>=11.0 Change bf16 to fp16 for non Ampere GPUs