nateraw commited on
Commit
137bd5e
·
1 Parent(s): eff8cb7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
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"),