Muhammad Taqi Raza
commited on
Commit
·
cc317e3
1
Parent(s):
3e33ccd
deleting demop vides
Browse files- inference_script.py +1 -1
inference_script.py
CHANGED
|
@@ -613,7 +613,7 @@ if __name__ == "__main__":
|
|
| 613 |
|
| 614 |
# if args.is_cpu_offload:
|
| 615 |
# pipe.enable_model_cpu_offload()
|
| 616 |
-
pipe.enable_sequential_cpu_offload()
|
| 617 |
# else:
|
| 618 |
# pipe.to("cuda")
|
| 619 |
|
|
|
|
| 613 |
|
| 614 |
# if args.is_cpu_offload:
|
| 615 |
# pipe.enable_model_cpu_offload()
|
| 616 |
+
# pipe.enable_sequential_cpu_offload()
|
| 617 |
# else:
|
| 618 |
# pipe.to("cuda")
|
| 619 |
|