dbaranchuk commited on
Commit
33a4a1f
·
verified ·
1 Parent(s): d5a87b2

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
  huggingface_token = os.getenv("HF_TOKEN")
14
  pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-large",
15
- torch_dtype=torch.bfloat16,
16
  token=huggingface_token)
17
  pipe = pipe.to("cuda")
18
 
 
12
 
13
  huggingface_token = os.getenv("HF_TOKEN")
14
  pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-large",
15
+ torch_dtype=torch.float16,
16
  token=huggingface_token)
17
  pipe = pipe.to("cuda")
18