Open dada-h opened 8 months ago
不知道方不方便添加一下对mps平台的支持？
InstantIDNode.py文件中： device = "cuda" if torch.cuda.is_available() else "cpu" 建议改为： from comfy.model_management import should_use_fp16, get_torch_device device = get_torch_device()
device = "cuda" if torch.cuda.is_available() else "cpu"
from comfy.model_management import should_use_fp16, get_torch_device
device = get_torch_device()
另外， torch_dtype=torch.float16 建议改为： torch_dtype=torch.float16 if should_use_fp16() else torch.float32 解决出黑图的问题
torch_dtype=torch.float16
torch_dtype=torch.float16 if should_use_fp16() else torch.float32
不知道方不方便添加一下对mps平台的支持？
InstantIDNode.py文件中：
device = "cuda" if torch.cuda.is_available() else "cpu"建议改为：from comfy.model_management import should_use_fp16, get_torch_devicedevice = get_torch_device()另外，
torch_dtype=torch.float16建议改为：torch_dtype=torch.float16 if should_use_fp16() else torch.float32解决出黑图的问题