kadirnar commited on
Commit
2bbfa11
1 Parent(s): a899214

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -28,12 +28,12 @@ def download_model(repo_id, model_name):
28
 
29
  import glob
30
 
31
- @spaces.GPU(duration=120)
32
  def run_inference(prompt_text):
33
  repo_id = "hpcai-tech/Open-Sora"
34
 
35
  # Map model names to their respective configuration files
36
- model_name = "OpenSora-v1-16x256x256.pth"
37
  config_mapping = {
38
  "OpenSora-v1-16x256x256.pth": "configs/opensora/inference/16x256x256.py",
39
  "OpenSora-v1-HQ-16x256x256.pth": "configs/opensora/inference/16x512x512.py",
 
28
 
29
  import glob
30
 
31
+ @spaces.GPU(duration=180)
32
  def run_inference(prompt_text):
33
  repo_id = "hpcai-tech/Open-Sora"
34
 
35
  # Map model names to their respective configuration files
36
+ model_name = "OpenSora-v1-HQ-16x256x256.pth"
37
  config_mapping = {
38
  "OpenSora-v1-16x256x256.pth": "configs/opensora/inference/16x256x256.py",
39
  "OpenSora-v1-HQ-16x256x256.pth": "configs/opensora/inference/16x512x512.py",