inoculatemedia commited on
Commit
a0e8e44
·
verified ·
1 Parent(s): 01f8320

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
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)