Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
56e8fae
1
Parent(s):
9015bd1
Update app.py
Browse files
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)
|