support patch ascend

2025-09-05 12:03:13 +08:00
parent ea0db79ebe
commit db19b64849
7 changed files with 77 additions and 29 deletions
--- a/patch.ascend/device.py.patch
+++ b/patch.ascend/device.py.patch
@@ -0,0 +1,2 @@
+27d26
+<     assert eles[0] in ['cpu', 'cuda', 'gpu'], err_msg
--- a/patch.ascend/text_to_video_synthesis_model.py.patch
+++ b/patch.ascend/text_to_video_synthesis_model.py.patch
@@ -0,0 +1,10 @@
+60a61
+>         print(f"kwargs: {kwargs}")
+62a64
+>         print(f"device: {self.device}")
+129c131
+<             layer='penultimate')
+---
+>             layer='penultimate', device=self.device)
+224a227
+>         print(f"self.device: {self.device}")