Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -85,7 +85,7 @@ pipeline = DistilledPipeline(
|
|
| 85 |
spatial_upsampler_path=SPATIAL_UPSAMPLER,
|
| 86 |
gemma_root=GEMMA_ROOT,
|
| 87 |
loras=[],
|
| 88 |
-
|
| 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=
|
|
|
|
| 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,
|