finetrainers/3dgs-dissolve
Viewer • Updated • 101 • 73 • 2
How to use finetrainers/Wan2.1-T2V-1.3B-3dgs-v0 with Diffusers:
pip install -U diffusers transformers accelerate
import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("finetrainers/Wan2.1-T2V-1.3B-3dgs-v0", dtype=torch.bfloat16, device_map="cuda")
prompt = "3DGS_DISSOLVE A lone sailor grips the helm as towering waves crash against the boat. Lightning splits the sky, illuminating the chaotic sea. The storm intensifies, but then, as if merging with the elements, the sailor's figure slowly transforms into a cascade of red sparks, scattering into the wind."
image = pipe(prompt).images[0]This is a LoRA fine-tune of the Wan-AI/Wan2.1-T2V-1.3B-Diffusers model on the finetrainers/3dgs-dissolve dataset.
Code: https://github.com/a-r-r-o-w/finetrainers
This is an experimental checkpoint and its poor generalization is well-known.
Inference code:
import torch
from diffusers import WanPipeline
from diffusers.utils import export_to_video
pipe = WanPipeline.from_pretrained(
"Wan-AI/Wan2.1-T2V-1.3B-Diffusers", torch_dtype=torch.bfloat16
).to("cuda")
pipe.load_lora_weights("finetrainers/Wan2.1-T2V-1.3B-3dgs-v0", adapter_name="wan-lora")
pipe.set_adapters(["wan-lora"], [0.5])
video = pipe("<my-awesome-prompt>").frames[0]
export_to_video(video, "output.mp4", fps=24)
Training logs are available on WandB here.
Base model
Wan-AI/Wan2.1-T2V-1.3B