update

a-r-r-o-w · a-r-r-o-w · commit 26a5a5c9b0fc · 2025-07-17T14:08:51.000+02:00
diff --git a/src/diffusers/models/_modeling_parallel.py b/src/diffusers/models/_modeling_parallel.py
@@ -1,4 +1,6 @@
-# Experimental parallelism support for Diffusers.
+# 🚨🚨🚨 Experimental parallelism support for Diffusers 🚨🚨🚨
+# Experimental changes are subject to change and APIs may break without warning.
+
 # Copyright 2025 The HuggingFace Team. All rights reserved.
 #
 # Licensed under the Apache License, Version 2.0 (the "License");
diff --git a/src/diffusers/models/attention_dispatch.py b/src/diffusers/models/attention_dispatch.py
@@ -21,7 +21,6 @@
 
 import torch
 import torch.distributed._functional_collectives as funcol
-import torch.distributed.tensor
 
 from ..utils import (
     get_logger,

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,6 @@`
`1`		`-# Experimental parallelism support for Diffusers.`
	`1`	`+# 🚨🚨🚨 Experimental parallelism support for Diffusers 🚨🚨🚨`
	`2`	`+# Experimental changes are subject to change and APIs may break without warning.`
	`3`	`+`
`2`	`4`	`# Copyright 2025 The HuggingFace Team. All rights reserved.`
`3`	`5`	`#`
`4`	`6`	`# Licensed under the Apache License, Version 2.0 (the "License");`