Spaces:
Running
on
Zero
Running
on
Zero
Commit
·
5f4dd46
1
Parent(s):
d2e9731
update app
Browse files
app.py
CHANGED
|
@@ -70,7 +70,7 @@ if lora_path:
|
|
| 70 |
)
|
| 71 |
print(f"✓ Configured scheduler (flow_shift=2.0)")
|
| 72 |
|
| 73 |
-
pipe.enable_model_cpu_offload()
|
| 74 |
pipe.to("cuda")
|
| 75 |
end = time.time()
|
| 76 |
print(f"Model loaded in {end - start:.2f}s.")
|
|
|
|
| 70 |
)
|
| 71 |
print(f"✓ Configured scheduler (flow_shift=2.0)")
|
| 72 |
|
| 73 |
+
# pipe.enable_model_cpu_offload()
|
| 74 |
pipe.to("cuda")
|
| 75 |
end = time.time()
|
| 76 |
print(f"Model loaded in {end - start:.2f}s.")
|