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