Fixing Pytorch 2.6.0+cu126 ＆xformers 0.0.29.post2 ＆α

2025年1月31日 15:38

Fixing Pytorch 2.6.0+cu126

Dev版の開発が進められていたPytorch 2.6.0+cu126が、遂に正式版で公開されたので、A1111、Forge、reForge、ComfyUI...その他Pytorchを使用する環境全てに対して更新をかけました。

ComfyUIではReactorノードにおいてエラーが起きる現象がありますが、この点に関しては暫定的な措置にはなりますが、以下記事で解決策を記載しています。

勿論、同時に「いつものやつ」は実施していますが…

一点だけ、想定外の事がありました。

Dev版の時点で、内装CUDAの更新方法については、一つ適用できるdllファイルについて変更があったのですが…

正式版になってみると、上記事の変更点は結局反映されていませんでした。反映させると、エラーを起こして動作不能に陥ります。

つまり要するに、今まで通り、という事です。

xformers 0.0.29.post2

対応が速いですね。早速、Pytorch2.6.0に対応してきました。

尚、xformersではなくcross-attentionでも、以下記事で私が作成している設定であればtorch.compileは動作します。inductorではなく、cudagraphsで動作させている為と思われます。

過去、作成してきた各WebUIのインストール方法についても、順次Pytorch2.6.0+cu126に合わせる形で更新していきます。

reForge ＆ Forge

Forgeのインストール方法は完全にreForgeと同じなので、以下記事を参照して下さい。

A1111 ＆ ComfyUI

ComfyUIに関しては前述した通り、Reactorでエラーが発生するので、対策を追記しています。

Re-fixed torch.compile

また、上記記事内で公開している、torch.compileを動作させる為の、global_config.pyの記述を変更しています。

import torch._dynamo
torch._dynamo.config.suppress_errors = True

# NODE_CLASS_MAPPINGS を追加
NODE_CLASS_MAPPINGS = {}

以下のエラーメッセージが発生している事に気づいた為、このエラーメッセージを消す為の修正です。

Skip D:\USERFILES\ComfyUI\ComfyUI\custom_nodes\global_config.py module for custom nodes due to the lack of NODE_CLASS_MAPPINGS.