Skip to content
代码片段 群组 项目
提交 b07258ce 编辑于 作者: comfyanonymous's avatar comfyanonymous
浏览文件

Fix typo.

Let me know if this slows things down on 2000 series and below.
上级 31e54b70
No related branches found
No related tags found
无相关合并请求
......@@ -1121,7 +1121,7 @@ def should_use_bf16(device=None, model_params=0, prioritize_performance=True, ma
bf16_works = torch.cuda.is_bf16_supported()
if bf16_works or manual_cast:
if bf16_works and manual_cast:
free_model_memory = maximum_vram_for_weights(device)
if (not prioritize_performance) or model_params * 4 > free_model_memory:
return True
......
0% 加载中 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册