Upload 2 files
Browse files- handler.py +1 -1
- requirements.txt +2 -2
handler.py
CHANGED
|
@@ -9,7 +9,7 @@ import torch
|
|
| 9 |
from torchao.quantization import quantize_, autoquant, int8_dynamic_activation_int8_weight
|
| 10 |
from huggingface_hub import hf_hub_download
|
| 11 |
|
| 12 |
-
IS_COMPILE =
|
| 13 |
IS_TURBO = False
|
| 14 |
|
| 15 |
if IS_COMPILE:
|
|
|
|
| 9 |
from torchao.quantization import quantize_, autoquant, int8_dynamic_activation_int8_weight
|
| 10 |
from huggingface_hub import hf_hub_download
|
| 11 |
|
| 12 |
+
IS_COMPILE = False
|
| 13 |
IS_TURBO = False
|
| 14 |
|
| 15 |
if IS_COMPILE:
|
requirements.txt
CHANGED
|
@@ -1,10 +1,10 @@
|
|
| 1 |
huggingface_hub
|
| 2 |
-
torch
|
| 3 |
torchvision
|
|
|
|
| 4 |
torchao==0.9.0
|
| 5 |
diffusers==0.32.2
|
| 6 |
peft
|
| 7 |
-
accelerate
|
| 8 |
transformers
|
| 9 |
numpy
|
| 10 |
scipy
|
|
|
|
| 1 |
huggingface_hub
|
| 2 |
+
torch==2.4.0
|
| 3 |
torchvision
|
| 4 |
+
torchaudio
|
| 5 |
torchao==0.9.0
|
| 6 |
diffusers==0.32.2
|
| 7 |
peft
|
|
|
|
| 8 |
transformers
|
| 9 |
numpy
|
| 10 |
scipy
|