Spaces:
Running
Running
Ming Li
commited on
Commit
•
ea0de4a
1
Parent(s):
7238e55
update requirements.txt
Browse files
model.py
CHANGED
@@ -59,7 +59,7 @@ class Model:
|
|
59 |
)
|
60 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
61 |
if self.device.type == "cuda":
|
62 |
-
pipe.
|
63 |
pipe.to(self.device)
|
64 |
torch.cuda.empty_cache()
|
65 |
gc.collect()
|
|
|
59 |
)
|
60 |
pipe.scheduler = UniPCMultistepScheduler.from_config(pipe.scheduler.config)
|
61 |
if self.device.type == "cuda":
|
62 |
+
pipe.disable_xformers_memory_efficient_attention()
|
63 |
pipe.to(self.device)
|
64 |
torch.cuda.empty_cache()
|
65 |
gc.collect()
|