Spaces:
Running
on
Zero
Running
on
Zero
test
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ CACHE_DIR = None
|
|
16 |
@spaces.GPU
|
17 |
def optimize_pipeline(pipeline):
|
18 |
pipeline.to(dtype=torch.float16)
|
19 |
-
pipeline.enable_xformers_memory_efficient_attention()
|
20 |
return pipeline
|
21 |
|
22 |
|
|
|
16 |
@spaces.GPU
|
17 |
def optimize_pipeline(pipeline):
|
18 |
pipeline.to(dtype=torch.float16)
|
19 |
+
# pipeline.enable_xformers_memory_efficient_attention()
|
20 |
return pipeline
|
21 |
|
22 |
|