macbook 类型不支持 float64 报错

[oatt123]

TypeError: Cannot convert a MPS Tensor to float64 dtype as the MPS framework doesn't support float64. Please use float32 instead.

File "/Applications/ComfyUI.app/Contents/Resources/ComfyUI/execution.py", line 525, in execute
output_data, output_ui, has_subgraph, has_pending_tasks = await get_output_data(prompt_id, unique_id, obj, input_data_all, execution_block_cb=execution_block_cb, pre_execute_cb=pre_execute_cb, v3_data=v3_data)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/Applications/ComfyUI.app/Contents/Resources/ComfyUI/execution.py", line 334, in get_output_data
return_values = await _async_map_node_over_list(prompt_id, unique_id, obj, input_data_all, obj.FUNCTION, allow_interrupt=True, execution_block_cb=execution_block_cb, pre_execute_cb=pre_execute_cb, v3_data=v3_data)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/Applications/ComfyUI.app/Contents/Resources/ComfyUI/execution.py", line 308, in _async_map_node_over_list
await process_inputs(input_dict, i)

File "/Applications/ComfyUI.app/Contents/Resources/ComfyUI/execution.py", line 296, in process_inputs
result = f(**inputs)
^^^^^^^^^^^

File "/Users/zhangjie/Documents/ComfyUI/custom_nodes/ComfyUI-FASHN-VTON/nodes.py", line 231, in process
result = pipeline(
^^^^^^^^^

File "/Users/zhangjie/Documents/ComfyUI/.venv/lib/python3.12/site-packages/torch/utils/_contextlib.py", line 124, in decorate_context
return func(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^

File "/Users/zhangjie/Documents/ComfyUI/custom_nodes/ComfyUI-FASHN-VTON/fashn_vton/pipeline.py", line 549, in call
images = self._sample(
^^^^^^^^^^^^^

File "/Users/zhangjie/Documents/ComfyUI/.venv/lib/python3.12/site-packages/torch/utils/_contextlib.py", line 124, in decorate_context
return func(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^

File "/Users/zhangjie/Documents/ComfyUI/custom_nodes/ComfyUI-FASHN-VTON/fashn_vton/pipeline.py", line 396, in _sample
pred = self.tryon_model.forward_for_cfg(images, t_vec, **model_kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/Users/zhangjie/Documents/ComfyUI/custom_nodes/ComfyUI-FASHN-VTON/fashn_vton/tryon_mmdit.py", line 471, in forward_for_cfg
all_logits = self.forward(*duplicated_args, **duplicated_kwargs)["x"]
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/Users/zhangjie/Documents/ComfyUI/custom_nodes/ComfyUI-FASHN-VTON/fashn_vton/tryon_mmdit.py", line 529, in forward
x_pe = self.pe_embedder(x_ids)
^^^^^^^^^^^^^^^^^^^^^^^

File "/Users/zhangjie/Documents/ComfyUI/.venv/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1779, in _wrapped_call_impl
return self._call_impl(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/Users/zhangjie/Documents/ComfyUI/.venv/lib/python3.12/site-packages/torch/nn/modules/module.py", line 1790, in _call_impl
return forward_call(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/Users/zhangjie/Documents/ComfyUI/custom_nodes/ComfyUI-FASHN-VTON/fashn_vton/tryon_mmdit.py", line 61, in forward
[rope(ids[..., i], self.axes_dim[i], self.theta) for i in range(n_axes)],
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^

File "/Users/zhangjie/Documents/ComfyUI/custom_nodes/ComfyUI-FASHN-VTON/fashn_vton/tryon_mmdit.py", line 35, in rope
scale = torch.arange(0, dim, 2, dtype=torch.float64, device=pos.device) / dim
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^