Spaces:
Running
on
Zero
Running
on
Zero
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)
|