Spaces:
Running
Running
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 |
|