ford442 commited on
Commit
7f76633
1 Parent(s): b00790e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -1
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": "800x512, 97 frames", "width": 768, "height": 768, "num_frames": 200},
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},