Spaces:
Runtime error
Runtime error
import torch | |
from diffusers import AnimateDiffPipeline, LCMScheduler, MotionAdapter | |
from diffusers.utils import export_to_gif | |
adapter = MotionAdapter.from_pretrained("wangfuyun/AnimateLCM", torch_dtype=torch.float16) | |
pipe = AnimateDiffPipeline.from_pretrained("emilianJR/epiCRealism", motion_adapter=adapter, torch_dtype=torch.float16) | |
pipe.scheduler = LCMScheduler.from_config(pipe.scheduler.config, beta_schedule="linear") | |
pipe.load_lora_weights("wangfuyun/AnimateLCM", weight_name="AnimateLCM_sd15_t2v_lora.safetensors", adapter_name="lcm-lora") | |
pipe.set_adapters(["lcm-lora"], [0.8]) | |
pipe.enable_vae_slicing() | |
pipe.enable_model_cpu_offload() | |
output = pipe( | |
prompt="A space rocket with trails of smoke behind it launching into space from the desert, 4k, high resolution", | |
negative_prompt="bad quality, worse quality, low resolution", | |
num_frames=16, | |
guidance_scale=2.0, | |
num_inference_steps=6, | |
generator=torch.Generator("cpu").manual_seed(0), | |
) | |
frames = output.frames[0] | |
export_to_gif(frames, "animatelcm.gif") | |