diffsynth-engine 0.4.3.dev4__py3-none-any.whl → 0.4.3.dev5__py3-none-any.whl

diffsynth_engine/utils/parallel.py

@@ -331,19 +331,14 @@ class ParallelWrapper:
         master_port: int = 29500,
         device: str = "cuda",
     ):
-        current_method = mp.get_start_method(allow_none=True)
-        if current_method is None or current_method != "spawn":
-            try:
-                mp.set_start_method("spawn")
-            except RuntimeError as e:
-                raise RuntimeError("Failed to set start method to spawn:", e)
         super().__init__()
         self.config = module.config if isinstance(module, BasePipeline) else None
         self._module_name = module.__class__.__name__
 
         self.world_size = cfg_degree * sp_ulysses_degree * sp_ring_degree * tp_degree
-        self.queue_in = mp.Queue()
-        self.queue_out = mp.Queue()
+        spawn_ctx = mp.get_context("spawn")
+        self.queue_in = spawn_ctx.Queue()
+        self.queue_out = spawn_ctx.Queue()
         self.ctx = mp.spawn(
             _worker_loop,
             args=(

{diffsynth_engine-0.4.3.dev4.dist-info → diffsynth_engine-0.4.3.dev5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: diffsynth_engine
-Version: 0.4.3.dev4
+Version: 0.4.3.dev5
 Author: MuseAI x ModelScope
 Classifier: Programming Language :: Python :: 3
 Classifier: Operating System :: OS Independent

{diffsynth_engine-0.4.3.dev4.dist-info → diffsynth_engine-0.4.3.dev5.dist-info}/RECORD

@@ -164,12 +164,12 @@ diffsynth_engine/utils/lock.py,sha256=1Ipgst9eEFfFdViAvD5bxdB6HnHHBcqWYOb__fGaPU
 diffsynth_engine/utils/logging.py,sha256=XB0xTT8PBN6btkOjFtOvjlrOCRVgDGT8PFAp1vmse28,467
 diffsynth_engine/utils/offload.py,sha256=8dHMEzAQcE-GNbFYotGQg2YBBB0eywoRTIffVmK4A0k,3746
 diffsynth_engine/utils/onnx.py,sha256=jeWUudJHnESjuiEAHyUZYUZz7dCj34O9aGjHCe8yjWo,1149
-diffsynth_engine/utils/parallel.py,sha256=TtB6FzP2qo4VQqSenVnV-ZaKmp9xHaIWJ8D1ZjHtukE,17064
+diffsynth_engine/utils/parallel.py,sha256=Z9jqCv4mLV4JyXR3uTHyv1rujPiKU8PSCbAfiN9jkPc,16818
 diffsynth_engine/utils/platform.py,sha256=2lXdw6YkqcRONCeT98n4cyg1Ii8Ybbyj2Ns72Se9tlk,496
 diffsynth_engine/utils/prompt.py,sha256=YItMchoVzsG6y-LB4vzzDUWrkhKRVlt1HfVhxZjSxMQ,280
 diffsynth_engine/utils/video.py,sha256=Ne0rd2lb59UT1q5EotpjlY7OT8F9oTCFDyo1ST77uoQ,1004
-diffsynth_engine-0.4.3.dev4.dist-info/licenses/LICENSE,sha256=x7aBqQuVI0IYnftgoTPI_A0I_rjdjPPQkjnU6N2nikM,11346
-diffsynth_engine-0.4.3.dev4.dist-info/METADATA,sha256=4D5ubLDy1tWA_n94e0QPdgTHQHOjFxEp-FM7mzS_ClE,1110
-diffsynth_engine-0.4.3.dev4.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-diffsynth_engine-0.4.3.dev4.dist-info/top_level.txt,sha256=6zgbiIzEHLbhgDKRyX0uBJOV3F6VnGGBRIQvSiYYn6w,17
-diffsynth_engine-0.4.3.dev4.dist-info/RECORD,,
+diffsynth_engine-0.4.3.dev5.dist-info/licenses/LICENSE,sha256=x7aBqQuVI0IYnftgoTPI_A0I_rjdjPPQkjnU6N2nikM,11346
+diffsynth_engine-0.4.3.dev5.dist-info/METADATA,sha256=hMlrgbZrIStdg2Sr4iIePwnxCltZ1rttaP_1I2gXflA,1110
+diffsynth_engine-0.4.3.dev5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+diffsynth_engine-0.4.3.dev5.dist-info/top_level.txt,sha256=6zgbiIzEHLbhgDKRyX0uBJOV3F6VnGGBRIQvSiYYn6w,17
+diffsynth_engine-0.4.3.dev5.dist-info/RECORD,,