dbaranchuk commited on
Commit
9dac327
·
verified ·
1 Parent(s): 03e661a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ 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.getenv("HF_READ_TOKEN"))
16
  pipe = pipe.to("cuda")
17
 
18
  distill_check = 'yresearch/swd-large-6-steps'
 
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'