Assert the text embeddings process successfully runs (threestudio-project#387)

johnbanq · web-flow · commit e254d87d2a9b · 2023-12-28T02:12:54.000+08:00
diff --git a/threestudio/models/prompt_processors/base.py b/threestudio/models/prompt_processors/base.py
@@ -379,6 +379,7 @@ def prepare_text_embeddings(self):
                 )
                 subprocess.start()
                 subprocess.join()
+                assert subprocess.exitcode == 0, "prompt embedding process failed!"
             else:
                 self.spawn_func(
                     self.cfg.pretrained_model_name_or_path,

Original file line number	Diff line number	Diff line change
`@@ -379,6 +379,7 @@ def prepare_text_embeddings(self):`
`379`	`379`	`)`
`380`	`380`	`subprocess.start()`
`381`	`381`	`subprocess.join()`
	`382`	`+ assert subprocess.exitcode == 0, "prompt embedding process failed!"`
`382`	`383`	`else:`
`383`	`384`	`self.spawn_func(`
`384`	`385`	`self.cfg.pretrained_model_name_or_path,`