123
#2
by
Pegasg
- opened
README.md
CHANGED
|
@@ -10,7 +10,7 @@ library_name: diffusers
|
|
| 10 |
<p align="center">
|
| 11 |
<a href="https://contentv.github.io">
|
| 12 |
<img
|
| 13 |
-
src="https://img.shields.io/badge/
|
| 14 |
alt="Project Page"
|
| 15 |
/>
|
| 16 |
</a>
|
|
@@ -59,6 +59,7 @@ Our open-source 8B model (based on Stable Diffusion 3.5 Large and Wan-VAE) achie
|
|
| 59 |
#### Recommended PyTorch Version
|
| 60 |
|
| 61 |
- GPU: torch >= 2.3.1 (CUDA >= 12.2)
|
|
|
|
| 62 |
|
| 63 |
#### Installation
|
| 64 |
|
|
@@ -73,6 +74,8 @@ pip3 install -r requirements.txt
|
|
| 73 |
```bash
|
| 74 |
## For GPU
|
| 75 |
python3 demo.py
|
|
|
|
|
|
|
| 76 |
```
|
| 77 |
|
| 78 |
## 📊 VBench
|
|
|
|
| 10 |
<p align="center">
|
| 11 |
<a href="https://contentv.github.io">
|
| 12 |
<img
|
| 13 |
+
src="https://img.shields.io/badge/Demo-Project Page-0A66C2?logo=googlechrome&logoColor=blue"
|
| 14 |
alt="Project Page"
|
| 15 |
/>
|
| 16 |
</a>
|
|
|
|
| 59 |
#### Recommended PyTorch Version
|
| 60 |
|
| 61 |
- GPU: torch >= 2.3.1 (CUDA >= 12.2)
|
| 62 |
+
- NPU: torch and torch-npu >= 2.1.0 (CANN >= 8.0.RC2). Please refer to [Ascend Extension for PyTorch](https://gitee.com/ascend/pytorch) for the installation of torch-npu.
|
| 63 |
|
| 64 |
#### Installation
|
| 65 |
|
|
|
|
| 74 |
```bash
|
| 75 |
## For GPU
|
| 76 |
python3 demo.py
|
| 77 |
+
## For NPU
|
| 78 |
+
USE_ASCEND_NPU=1 python3 demo.py
|
| 79 |
```
|
| 80 |
|
| 81 |
## 📊 VBench
|