Muhammad Taqi Raza
commited on
Commit
·
ef1737d
1
Parent(s):
bd4af33
add app files
Browse files- inference_script.py +2 -2
inference_script.py
CHANGED
|
@@ -614,8 +614,8 @@ if __name__ == "__main__":
|
|
| 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 |
|
| 620 |
if args.is_vae_st:
|
| 621 |
pipe.vae.enable_slicing()
|
|
|
|
| 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 |
|
| 620 |
if args.is_vae_st:
|
| 621 |
pipe.vae.enable_slicing()
|