Improvements for Wan 2.2 support

- add xet support and add the xet cache to manageable directories
 - xet is enabled by default
 - fix logging to root in various places
 - improve logging about model unloading and loading
 - TorchCompileNode now supports the VAE
 - torchaudio missing will cause less noise in the logs
 - feature flags will assume to be supporting everything in the distributed progress context
 - fixes progress notifications
This commit is contained in:
doctorpangloss
2025-07-28 14:36:27 -07:00
parent b5a50301f6
commit 03e5430121
19 changed files with 192 additions and 82 deletions
+5 -1
View File
@@ -43,7 +43,8 @@ def get_connection_feature(
def supports_feature(
sockets_metadata: Dict[str, Dict[str, Any]],
sid: str,
feature_name: str
feature_name: str,
force=True,
) -> bool:
"""
Check if a connection supports a specific feature.
@@ -52,10 +53,13 @@ def supports_feature(
sockets_metadata: Dictionary of socket metadata
sid: Session ID of the connection
feature_name: Name of the feature to check
force (bool): If it cannot be determined, assume True
Returns:
Boolean indicating if feature is supported
"""
if sockets_metadata is None or "__unimplemented" in sockets_metadata:
return force
return get_connection_feature(sockets_metadata, sid, feature_name, False) is True