multimodalart HF Staff commited on
Commit
b5c82c0
·
verified ·
1 Parent(s): 0067831

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -85,7 +85,7 @@ pipeline = DistilledPipeline(
85
  spatial_upsampler_path=SPATIAL_UPSAMPLER,
86
  gemma_root=GEMMA_ROOT,
87
  loras=[],
88
- # quantization=QuantizationPolicy.fp8_cast(),
89
  )
90
 
91
  # Preload all models so first request is fast.
@@ -112,7 +112,8 @@ ledger.spatial_upsampler = lambda: _spatial_upsampler
112
  print("All models preloaded!")
113
 
114
 
115
- @spaces.GPU(duration=120, size='xlarge')
 
116
  def generate_video(
117
  input_image,
118
  prompt: str,
 
85
  spatial_upsampler_path=SPATIAL_UPSAMPLER,
86
  gemma_root=GEMMA_ROOT,
87
  loras=[],
88
+ quantization=QuantizationPolicy.fp8_cast(),
89
  )
90
 
91
  # Preload all models so first request is fast.
 
112
  print("All models preloaded!")
113
 
114
 
115
+ @spaces.GPU(duration=300)
116
+ @torch.inference_mode()
117
  def generate_video(
118
  input_image,
119
  prompt: str,