Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -87,6 +87,7 @@ print('Quantizing, please wait...')
|
|
| 87 |
quantize(pipe.transformer, qfloat8)
|
| 88 |
freeze(pipe.transformer)
|
| 89 |
print('Model quantized!')
|
|
|
|
| 90 |
|
| 91 |
|
| 92 |
# Load 3D generation models
|
|
|
|
| 87 |
quantize(pipe.transformer, qfloat8)
|
| 88 |
freeze(pipe.transformer)
|
| 89 |
print('Model quantized!')
|
| 90 |
+
pipe.enable_model_cpu_offload()
|
| 91 |
|
| 92 |
|
| 93 |
# Load 3D generation models
|