import torch
from diffusers import DiffusionPipeline
# switch to "mps" for apple devices
pipe = DiffusionPipeline.from_pretrained("wav/TemporalNet2", dtype=torch.bfloat16, device_map="cuda")
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
image = pipe(prompt).images[0]TemporalNet2 for Diffusers
CiaraRowles/TemporalNet2 but converted for use with Diffusers!
At the moment loading the checkpoint requires a tiny 3-line fix to Diffusers (see this PR). You can install the working version of diffusers with:
pip install git+https://github.com/JCBrouwer/diffusers@expose-controlnet-conditioning-channels
Then you can use the script included in this repo to stylize videos using text prompts.
python temporalvideo_hf.py --help
- Downloads last month
- 10
Inference Providers NEW
This model isn't deployed by any Inference Provider. 🙋 Ask for provider support