mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-28 23:30:16 +08:00
fix torch 2.2 missing is_compiling member
This commit is contained in:
parent
1fd1c2c7fc
commit
2c7f8788f8
@ -1202,7 +1202,7 @@ def get_offload_stream(device):
|
||||
if NUM_STREAMS == 0:
|
||||
return None
|
||||
|
||||
if torch.compiler.is_compiling():
|
||||
if hasattr(torch.compiler, "is_compiling") and torch.compiler.is_compiling(): # pylint: disable=no-member
|
||||
return None
|
||||
|
||||
if device in STREAMS:
|
||||
|
||||
Loading…
Reference in New Issue
Block a user