Upload 2 files
Browse files- handler.py +1 -1
- requirements.txt +1 -5
handler.py
CHANGED
|
@@ -10,7 +10,7 @@ from torchao.quantization import quantize_, autoquant, int8_dynamic_activation_i
|
|
| 10 |
from torchao.quantization.quant_api import PerRow
|
| 11 |
from diffusers import FluxPipeline, FluxTransformer2DModel, AutoencoderKL, TorchAoConfig
|
| 12 |
|
| 13 |
-
IS_NEW_GPU =
|
| 14 |
IS_COMPILE = True
|
| 15 |
IS_TURBO = False
|
| 16 |
IS_4BIT = False
|
|
|
|
| 10 |
from torchao.quantization.quant_api import PerRow
|
| 11 |
from diffusers import FluxPipeline, FluxTransformer2DModel, AutoencoderKL, TorchAoConfig
|
| 12 |
|
| 13 |
+
IS_NEW_GPU = True
|
| 14 |
IS_COMPILE = True
|
| 15 |
IS_TURBO = False
|
| 16 |
IS_4BIT = False
|
requirements.txt
CHANGED
|
@@ -1,12 +1,9 @@
|
|
| 1 |
--extra-index-url https://download.pytorch.org/whl/cu124
|
| 2 |
-
#torch==2.5.1+cu124
|
| 3 |
-
#https://download.pytorch.org/whl/nightly/cu124/torch-2.7.0.dev20250306%2Bcu124-cp311-cp311-manylinux_2_28_x86_64.whl
|
| 4 |
torch>=2.6.0
|
| 5 |
torchvision
|
| 6 |
torchaudio
|
| 7 |
huggingface_hub
|
| 8 |
torchao>=0.9.0
|
| 9 |
-
#git+https://github.com/pytorch/ao
|
| 10 |
#diffusers==0.32.2
|
| 11 |
git+https://github.com/huggingface/diffusers
|
| 12 |
peft
|
|
@@ -16,5 +13,4 @@ scipy
|
|
| 16 |
Pillow
|
| 17 |
sentencepiece
|
| 18 |
protobuf
|
| 19 |
-
triton
|
| 20 |
-
#git+https://github.com/triton-lang/triton
|
|
|
|
| 1 |
--extra-index-url https://download.pytorch.org/whl/cu124
|
|
|
|
|
|
|
| 2 |
torch>=2.6.0
|
| 3 |
torchvision
|
| 4 |
torchaudio
|
| 5 |
huggingface_hub
|
| 6 |
torchao>=0.9.0
|
|
|
|
| 7 |
#diffusers==0.32.2
|
| 8 |
git+https://github.com/huggingface/diffusers
|
| 9 |
peft
|
|
|
|
| 13 |
Pillow
|
| 14 |
sentencepiece
|
| 15 |
protobuf
|
| 16 |
+
triton
|
|
|