ford442 commited on
Commit
f9c0d35
1 Parent(s): e6ea6a6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -7
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 = 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")
@@ -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": 140},
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},