Spaces:
Running
on
Zero
Running
on
Zero
Update pipeline_qwenimage_edit.py
Browse files
pipeline_qwenimage_edit.py
CHANGED
|
@@ -301,7 +301,7 @@ class QwenImageEditPipeline(DiffusionPipeline, QwenImageLoraLoaderMixin):
|
|
| 301 |
dtype: Optional[torch.dtype] = None,
|
| 302 |
):
|
| 303 |
# 确保设备一致性(ZeroGPU 修复)
|
| 304 |
-
self._ensure_device_consistency()
|
| 305 |
|
| 306 |
# 获取 text_encoder 的实际设备
|
| 307 |
encoder_device = next(self.text_encoder.parameters()).device
|
|
@@ -746,7 +746,7 @@ class QwenImageEditPipeline(DiffusionPipeline, QwenImageLoraLoaderMixin):
|
|
| 746 |
returning a tuple, the first element is a list with the generated images.
|
| 747 |
"""
|
| 748 |
# ZeroGPU 修复:确保设备一致性
|
| 749 |
-
self._ensure_device_consistency()
|
| 750 |
|
| 751 |
if not isinstance(images, (list, tuple)):
|
| 752 |
images = [images]
|
|
|
|
| 301 |
dtype: Optional[torch.dtype] = None,
|
| 302 |
):
|
| 303 |
# 确保设备一致性(ZeroGPU 修复)
|
| 304 |
+
# self._ensure_device_consistency()
|
| 305 |
|
| 306 |
# 获取 text_encoder 的实际设备
|
| 307 |
encoder_device = next(self.text_encoder.parameters()).device
|
|
|
|
| 746 |
returning a tuple, the first element is a list with the generated images.
|
| 747 |
"""
|
| 748 |
# ZeroGPU 修复:确保设备一致性
|
| 749 |
+
# self._ensure_device_consistency()
|
| 750 |
|
| 751 |
if not isinstance(images, (list, tuple)):
|
| 752 |
images = [images]
|