yerang commited on
Commit
2532f22
1 Parent(s): 8f011bd

Update flux_dev.py

Browse files
Files changed (1) hide show
  1. flux_dev.py +3 -3
flux_dev.py CHANGED
@@ -7,10 +7,10 @@ from diffusers import DiffusionPipeline, FlowMatchEulerDiscreteScheduler, Autoen
7
  from transformers import CLIPTextModel, CLIPTokenizer,T5EncoderModel, T5TokenizerFast
8
  from live_preview_helpers import calculate_shift, retrieve_timesteps, flux_pipe_call_that_returns_an_iterable_of_images
9
 
10
- #from huggingface_hub import login
11
  import os
12
- #token = os.getenv("HF_TOKEN")
13
- #login(token=token)
14
 
15
  dtype = torch.bfloat16
16
  device = "cuda" if torch.cuda.is_available() else "cpu"
 
7
  from transformers import CLIPTextModel, CLIPTokenizer,T5EncoderModel, T5TokenizerFast
8
  from live_preview_helpers import calculate_shift, retrieve_timesteps, flux_pipe_call_that_returns_an_iterable_of_images
9
 
10
+ from huggingface_hub import login
11
  import os
12
+ token = os.getenv("HF_TOKEN")
13
+ login(token=token)
14
 
15
  dtype = torch.bfloat16
16
  device = "cuda" if torch.cuda.is_available() else "cpu"