Spaces:
Runtime error
Runtime error
from stable_diffusion_videos import StableDiffusionWalkPipeline, Interface | |
from diffusers.schedulers import LMSDiscreteScheduler | |
import torch | |
import os | |
from huggingface_hub import HfFolder | |
HfFolder().save_token(os.environ['HF_TOKEN']) | |
pipeline = StableDiffusionWalkPipeline.from_pretrained( | |
"CompVis/stable-diffusion-v1-4", | |
torch_dtype=torch.float16, | |
revision="fp16", | |
scheduler=LMSDiscreteScheduler( | |
beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear" | |
) | |
).to("cuda") | |
interface = Interface(pipeline) | |
if __name__ == '__main__': | |
interface.launch() |