Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
|
@@ -82,6 +82,9 @@ text_encoder_2.requires_grad_(False)
|
|
| 82 |
image_encoder.requires_grad_(False)
|
| 83 |
transformer.requires_grad_(False)
|
| 84 |
|
|
|
|
|
|
|
|
|
|
| 85 |
if not high_vram:
|
| 86 |
# DynamicSwapInstaller is same as huggingface's enable_sequential_offload but 3x faster
|
| 87 |
DynamicSwapInstaller.install_model(transformer, device=gpu)
|
|
|
|
| 82 |
image_encoder.requires_grad_(False)
|
| 83 |
transformer.requires_grad_(False)
|
| 84 |
|
| 85 |
+
@spaces.GPU(duration=75)
|
| 86 |
+
|
| 87 |
+
|
| 88 |
if not high_vram:
|
| 89 |
# DynamicSwapInstaller is same as huggingface's enable_sequential_offload but 3x faster
|
| 90 |
DynamicSwapInstaller.install_model(transformer, device=gpu)
|