Compare commits
2 Commits
24742d2267
...
feat/mlu
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
9b0f87d152 | ||
|
|
cbef668d79 |
@@ -1,4 +1,4 @@
|
|||||||
FROM git.modelhub.org.cn:9443/enginex-cambricon/mlu370-pytorch:v25.01-torch2.5.0-torchmlu1.24.1-ubuntu22.04-py310
|
FROM mlu370-pytorch:v25.01-torch2.5.0-torchmlu1.24.1-ubuntu22.04-py310
|
||||||
|
|
||||||
WORKDIR /workspace
|
WORKDIR /workspace
|
||||||
ENV PATH=/torch/venv3/pytorch_infer/bin:/workspace/ffmpeg-mlu-v4.2.0/install/bin:/usr/local/neuware/bin:/usr/local/openmpi/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
|
ENV PATH=/torch/venv3/pytorch_infer/bin:/workspace/ffmpeg-mlu-v4.2.0/install/bin:/usr/local/neuware/bin:/usr/local/openmpi/bin:/usr/local/sbin:/usr/local/bin:/usr/sbin:/usr/bin:/sbin:/bin
|
||||||
|
|||||||
@@ -4,6 +4,7 @@
|
|||||||
```bash
|
```bash
|
||||||
docker build -t text2video:v0.1 .
|
docker build -t text2video:v0.1 .
|
||||||
```
|
```
|
||||||
|
其中基础镜像 mlu370-pytorch:v25.01-torch2.5.0-torchmlu1.24.1-ubuntu22.04-py310 联系寒武纪厂商技术支持可获取
|
||||||
|
|
||||||
### 模型下载
|
### 模型下载
|
||||||
模型地址:https://modelscope.cn/models/iic/text-to-video-synthesis
|
模型地址:https://modelscope.cn/models/iic/text-to-video-synthesis
|
||||||
|
|||||||
2
main.py
2
main.py
@@ -147,7 +147,7 @@ def main():
|
|||||||
|
|
||||||
print(f"\nAll done. vidoes: {len(records)}, total_elapsed: {total_elapsed:.3f}s, avg_latency: {avg_latency:.3f}")
|
print(f"\nAll done. vidoes: {len(records)}, total_elapsed: {total_elapsed:.3f}s, avg_latency: {avg_latency:.3f}")
|
||||||
print(f"Results JSON: {results_path}")
|
print(f"Results JSON: {results_path}")
|
||||||
print(f"Images dir : {out_dir.resolve()}")
|
print(f"Output dir : {out_dir.resolve()}")
|
||||||
|
|
||||||
|
|
||||||
if __name__ == "__main__":
|
if __name__ == "__main__":
|
||||||
|
|||||||
Reference in New Issue
Block a user