Spaces:
Sleeping
Sleeping
Commit
·
5120359
1
Parent(s):
9e112ae
no gr.noreload and preprocessor in the top
Browse files
app.py
CHANGED
|
@@ -27,6 +27,9 @@ API_KEY = os.environ.get("API_KEY", None)
|
|
| 27 |
print("CUDA version:", torch.version.cuda)
|
| 28 |
print("loading pipe")
|
| 29 |
compiled = False
|
|
|
|
|
|
|
|
|
|
| 30 |
# api = HfApi()
|
| 31 |
|
| 32 |
import spaces
|
|
@@ -99,9 +102,6 @@ def init(pipe):
|
|
| 99 |
pipe.enable_xformers_memory_efficient_attention()
|
| 100 |
pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
|
| 101 |
pipe.unet.set_attn_processor(AttnProcessor2_0())
|
| 102 |
-
from preprocess import Preprocessor
|
| 103 |
-
preprocessor = Preprocessor()
|
| 104 |
-
preprocessor.load("NormalBae")
|
| 105 |
print("Model Compiled!")
|
| 106 |
init(pipe)
|
| 107 |
|
|
|
|
| 27 |
print("CUDA version:", torch.version.cuda)
|
| 28 |
print("loading pipe")
|
| 29 |
compiled = False
|
| 30 |
+
from preprocess import Preprocessor
|
| 31 |
+
preprocessor = Preprocessor()
|
| 32 |
+
preprocessor.load("NormalBae")
|
| 33 |
# api = HfApi()
|
| 34 |
|
| 35 |
import spaces
|
|
|
|
| 102 |
pipe.enable_xformers_memory_efficient_attention()
|
| 103 |
pipe.unet = torch.compile(pipe.unet, mode="reduce-overhead", fullgraph=True)
|
| 104 |
pipe.unet.set_attn_processor(AttnProcessor2_0())
|
|
|
|
|
|
|
|
|
|
| 105 |
print("Model Compiled!")
|
| 106 |
init(pipe)
|
| 107 |
|