Update app.py
Browse files
app.py
CHANGED
@@ -25,6 +25,15 @@ import gc
|
|
25 |
import csv
|
26 |
from datetime import datetime
|
27 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
28 |
# Load Hugging Face token if needed
|
29 |
hf_token = os.getenv("HF_TOKEN")
|
30 |
openai_api_key = os.getenv("OPENAI_API_KEY")
|
@@ -149,7 +158,7 @@ preset_options = [
|
|
149 |
{"label": "896x608, 73 frames", "width": 896, "height": 608, "num_frames": 73},
|
150 |
{"label": "896x544, 81 frames", "width": 896, "height": 544, "num_frames": 81},
|
151 |
{"label": "832x544, 89 frames", "width": 832, "height": 544, "num_frames": 89},
|
152 |
-
{"label": "
|
153 |
{"label": "768x512, 97 frames", "width": 768, "height": 512, "num_frames": 97},
|
154 |
{"label": "800x480, 105 frames", "width": 800, "height": 480, "num_frames": 105},
|
155 |
{"label": "736x480, 113 frames", "width": 736, "height": 480, "num_frames": 113},
|
|
|
25 |
import csv
|
26 |
from datetime import datetime
|
27 |
|
28 |
+
torch.backends.cuda.matmul.allow_tf32 = True
|
29 |
+
torch.backends.cuda.matmul.allow_bf16_reduced_precision_reduction = True
|
30 |
+
torch.backends.cuda.matmul.allow_fp16_reduced_precision_reduction = True
|
31 |
+
torch.backends.cudnn.allow_tf32 = True
|
32 |
+
torch.backends.cudnn.deterministic = False
|
33 |
+
torch.backends.cudnn.benchmark = True
|
34 |
+
torch.set_float32_matmul_precision("highest")
|
35 |
+
|
36 |
+
|
37 |
# Load Hugging Face token if needed
|
38 |
hf_token = os.getenv("HF_TOKEN")
|
39 |
openai_api_key = os.getenv("OPENAI_API_KEY")
|
|
|
158 |
{"label": "896x608, 73 frames", "width": 896, "height": 608, "num_frames": 73},
|
159 |
{"label": "896x544, 81 frames", "width": 896, "height": 544, "num_frames": 81},
|
160 |
{"label": "832x544, 89 frames", "width": 832, "height": 544, "num_frames": 89},
|
161 |
+
{"label": "768x768, 200 frames", "width": 768, "height": 768, "num_frames": 160},
|
162 |
{"label": "768x512, 97 frames", "width": 768, "height": 512, "num_frames": 97},
|
163 |
{"label": "800x480, 105 frames", "width": 800, "height": 480, "num_frames": 105},
|
164 |
{"label": "736x480, 113 frames", "width": 736, "height": 480, "num_frames": 113},
|