multimodalart HF staff commited on
Commit
56e8fae
1 Parent(s): 9015bd1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -10,6 +10,7 @@ from PIL import Image, ImageDraw
10
  import numpy as np
11
  import spaces
12
  from huggingface_hub import hf_hub_download
 
13
 
14
  # Load fp8
15
  #transformer = FluxTransformer2DModel.from_single_file("https://huggingface.co/Kijai/flux-fp8/blob/main/flux1-dev-fp8.safetensors", torch_dtype=torch.bfloat16)
 
10
  import numpy as np
11
  import spaces
12
  from huggingface_hub import hf_hub_download
13
+ from optimum.quanto import freeze, qfloat8, quantize
14
 
15
  # Load fp8
16
  #transformer = FluxTransformer2DModel.from_single_file("https://huggingface.co/Kijai/flux-fp8/blob/main/flux1-dev-fp8.safetensors", torch_dtype=torch.bfloat16)