Update README.md
Browse files
README.md
CHANGED
|
@@ -1,33 +1,10 @@
|
|
| 1 |
|
| 2 |
|
| 3 |
## Notice
|
| 4 |
-
* This is a diffusers
|
| 5 |
-
|
| 6 |
|
| 7 |
|
| 8 |
-
```
|
| 9 |
-
from diffusers import GGUFQuantizationConfig,WanTransformer3DModel,WanVideoToVideoPipeline
|
| 10 |
-
from transformers import UMT5EncoderModel
|
| 11 |
-
from diffusers.models import AutoencoderKLWan
|
| 12 |
-
|
| 13 |
-
gguf_path="https://huggingface.co/smthem/SteadyDancer-14B-gguf/blob/main/SteadyDancer-14B-Q8_0.gguf"
|
| 14 |
-
model_id="Wan-AI/Wan2.1-I2V-14B-720P-Diffusers"
|
| 15 |
-
|
| 16 |
-
transformer = WanTransformer3DModel.from_single_file(
|
| 17 |
-
gguf_path,
|
| 18 |
-
config=model_id,
|
| 19 |
-
quantization_config=GGUFQuantizationConfig(compute_dtype=torch.bfloat16),
|
| 20 |
-
torch_dtype=torch.bfloat16,
|
| 21 |
-
)
|
| 22 |
-
|
| 23 |
-
vae=AutoencoderKLWan.from_pretrained(model_id, torch_dtype=torch.bfloat16,)
|
| 24 |
-
text_encoder=UMT5EncoderModel.from_pretrained(model_id, torch_dtype=torch.bfloat16,)
|
| 25 |
-
|
| 26 |
-
pipe = WanVideoToVideoPipeline.from_pretrained(model_id, vae=vae,transformer=transformer,text_encoder=text_encoder, torch_dtype=torch.bfloat16)
|
| 27 |
-
|
| 28 |
-
# run infer
|
| 29 |
-
...
|
| 30 |
-
|
| 31 |
```
|
| 32 |
|
| 33 |
|
|
|
|
| 1 |
|
| 2 |
|
| 3 |
## Notice
|
| 4 |
+
* This is a diffusers test model ,只是测试模型
|
| 5 |
+
* gguf 的量化偏差导致最终输出结果太差,还是用safetensor吧
|
| 6 |
|
| 7 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
```
|
| 9 |
|
| 10 |
|