multimodalart HF staff commited on
Commit
9015bd1
1 Parent(s): 70e5835

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -21,7 +21,7 @@ from huggingface_hub import hf_hub_download
21
  #quantize(controlnet, weights=qfloat8)
22
  #freeze(controlnet)
23
 
24
- text_encoder_2 = T5EncoderModel.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="text_encoder_2", torch_dtype=dtype)
25
  quantize(text_encoder_2, weights=qfloat8)
26
  freeze(text_encoder_2)
27
 
 
21
  #quantize(controlnet, weights=qfloat8)
22
  #freeze(controlnet)
23
 
24
+ text_encoder_2 = T5EncoderModel.from_pretrained("black-forest-labs/FLUX.1-dev", subfolder="text_encoder_2", torch_dtype=torch.bfloat16)
25
  quantize(text_encoder_2, weights=qfloat8)
26
  freeze(text_encoder_2)
27