adapted to ms2.7

Author: QiuYuFong

examples/opensora_hpcai/opensora/models/layers/operation_selector.py

@@ -1,6 +1,6 @@
 import mindspore as ms
 from mindspore import mint, ops
-from mindspore.ops.function.array_func import chunk_ext, repeat_interleave_ext
+from mindspore.ops.function.array_func import repeat_interleave_ext
 
 use_dynamic_ops = False
 
@@ -66,7 +66,7 @@ def get_chunk_op():
     if (mode == 0) and (not check_dynamic_mode()):
         return ops.chunk
     else:
-        return chunk_ext
+        return mint.chunk
 
 
 def get_split_op():

examples/opensora_hpcai/opensora/models/layers/rotary_embedding.py

@@ -12,15 +12,14 @@
 import numpy as np
 
 import mindspore as ms
-from mindspore import Parameter, Tensor, dtype, nn, ops
-from mindspore.ops.function.array_func import chunk_ext as chunk
+from mindspore import Parameter, Tensor, dtype, mint, nn, ops
 
 from .operation_selector import get_repeat_interleave_op
 
 
 def rotate_half(x: Tensor) -> Tensor:
     x = x.reshape(x.shape[:-1] + (-1, 2))  # ... (d r) -> ... d r, r = 2
-    x1, x2 = chunk(x, 2, -1)
+    x1, x2 = mint.chunk(x, 2, -1)
     x = ops.concat((-x2, x1), axis=-1)
     return x.reshape(x.shape[:-2] + (-1,))  # '... d r -> ... (d r)'
 

examples/opensora_hpcai/scripts/inference.py

@@ -478,7 +478,8 @@ def main(args):
                 videos.append(to_numpy(samples)[:, args.condition_frame_length if loop_i > 0 else 0 :])
             batch_time = time.time() - start_time
             logger.info(
-                f"Batch time cost: {batch_time:.3f}s, sampling speed: {args.sampling_steps * ns / batch_time:.4f} step/s"
+                f"Batch time cost: {batch_time:.3f}s, sampling speed: "
+                f"{batch_time / (args.sampling_steps * ns):.4f} s/step"
             )
 
         latents = np.concatenate(latents, axis=2)