alex
commited on
Commit
·
a381f1e
1
Parent(s):
3a03985
duration fixed
Browse files
app.py
CHANGED
|
@@ -39,15 +39,6 @@ import torch
|
|
| 39 |
print(f"Torch version: {torch.__version__}")
|
| 40 |
print(f"FlashAttention available: {flash_attention_installed}")
|
| 41 |
|
| 42 |
-
|
| 43 |
-
try:
|
| 44 |
-
import flash_attn_interface
|
| 45 |
-
FLASH_ATTN_3_AVAILABLE = True
|
| 46 |
-
print(f'FLASH_ATTN_3_AVAILABLE-innit:{FLASH_ATTN_3_AVAILABLE}')
|
| 47 |
-
except ModuleNotFoundError:
|
| 48 |
-
print(f'faield FLASH_ATTN_3_AVAILABLE-innit:{FLASH_ATTN_3_AVAILABLE}')
|
| 49 |
-
FLASH_ATTN_3_AVAILABLE = False
|
| 50 |
-
|
| 51 |
import gradio as gr
|
| 52 |
import argparse
|
| 53 |
from ovi.ovi_fusion_engine import OviFusionEngine, DEFAULT_CONFIG
|
|
@@ -120,7 +111,7 @@ if use_image_gen:
|
|
| 120 |
print("loaded model")
|
| 121 |
|
| 122 |
|
| 123 |
-
@spaces.GPU()
|
| 124 |
def generate_video(
|
| 125 |
text_prompt,
|
| 126 |
image,
|
|
|
|
| 39 |
print(f"Torch version: {torch.__version__}")
|
| 40 |
print(f"FlashAttention available: {flash_attention_installed}")
|
| 41 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 42 |
import gradio as gr
|
| 43 |
import argparse
|
| 44 |
from ovi.ovi_fusion_engine import OviFusionEngine, DEFAULT_CONFIG
|
|
|
|
| 111 |
print("loaded model")
|
| 112 |
|
| 113 |
|
| 114 |
+
@spaces.GPU(duration=120)
|
| 115 |
def generate_video(
|
| 116 |
text_prompt,
|
| 117 |
image,
|