dbaranchuk commited on
Commit
09b5182
·
verified ·
1 Parent(s): 9dac327

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -10,9 +10,11 @@ from diffusers import DiffusionPipeline
10
  from inference import run
11
  from peft import LoraConfig, get_peft_model, PeftModel
12
 
 
 
13
  pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-large",
14
  torch_dtype=torch.bfloat16,
15
- token=os.environ.get("HF_TOKEN"))
16
  pipe = pipe.to("cuda")
17
 
18
  distill_check = 'yresearch/swd-large-6-steps'
 
10
  from inference import run
11
  from peft import LoraConfig, get_peft_model, PeftModel
12
 
13
+ huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
14
+ print(huggingface_token)
15
  pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-large",
16
  torch_dtype=torch.bfloat16,
17
+ token=huggingface_token)
18
  pipe = pipe.to("cuda")
19
 
20
  distill_check = 'yresearch/swd-large-6-steps'