OrlandoHugBot commited on
Commit
ed53052
·
verified ·
1 Parent(s): bdba128

Update pipeline_qwenimage_edit.py

Browse files
Files changed (1) hide show
  1. pipeline_qwenimage_edit.py +2 -2
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]