jiuface commited on
Commit
3fc0dd0
1 Parent(s): 6373ff8
__pycache__/live_preview_helpers.cpython-310.pyc ADDED
Binary file (4.03 kB). View file
 
app.py CHANGED
@@ -9,12 +9,15 @@ import json
9
  import logging
10
  from diffusers import DiffusionPipeline, AutoencoderTiny, AutoencoderKL
11
  from live_preview_helpers import calculate_shift, retrieve_timesteps, flux_pipe_call_that_returns_an_iterable_of_images
12
-
13
  from huggingface_hub import hf_hub_download, HfFileSystem, ModelCard, snapshot_download
14
  import copy
15
  import random
16
  import time
17
 
 
 
 
18
 
19
  # init
20
  dtype = torch.bfloat16
 
9
  import logging
10
  from diffusers import DiffusionPipeline, AutoencoderTiny, AutoencoderKL
11
  from live_preview_helpers import calculate_shift, retrieve_timesteps, flux_pipe_call_that_returns_an_iterable_of_images
12
+ from huggingface_hub import login
13
  from huggingface_hub import hf_hub_download, HfFileSystem, ModelCard, snapshot_download
14
  import copy
15
  import random
16
  import time
17
 
18
+ HF_TOKEN = os.environ.get("HF_TOKEN")
19
+
20
+ login(token=HF_TOKEN)
21
 
22
  # init
23
  dtype = torch.bfloat16