Update app.py
Browse files
app.py
CHANGED
@@ -18,6 +18,10 @@ from diffusers.schedulers import LMSDiscreteScheduler
|
|
18 |
from matplotlib import pyplot as plt
|
19 |
from stable_diffusion_videos import StableDiffusionWalkPipeline, generate_images, get_timesteps_arr
|
20 |
|
|
|
|
|
|
|
|
|
21 |
pipe = StableDiffusionWalkPipeline.from_pretrained(
|
22 |
"runwayml/stable-diffusion-v1-5",
|
23 |
vae=AutoencoderKL.from_pretrained(f"stabilityai/sd-vae-ft-ema"),
|
|
|
18 |
from matplotlib import pyplot as plt
|
19 |
from stable_diffusion_videos import StableDiffusionWalkPipeline, generate_images, get_timesteps_arr
|
20 |
|
21 |
+
from huggingface_hub import HfFolder
|
22 |
+
|
23 |
+
HfFolder().save_token(os.environ['HF_TOKEN'])
|
24 |
+
|
25 |
pipe = StableDiffusionWalkPipeline.from_pretrained(
|
26 |
"runwayml/stable-diffusion-v1-5",
|
27 |
vae=AutoencoderKL.from_pretrained(f"stabilityai/sd-vae-ft-ema"),
|