1inkusFace commited on
Commit
43ff127
·
verified ·
1 Parent(s): 24d0322

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -11,7 +11,7 @@ from skyreelsinfer.skyreels_video_infer import SkyReelsVideoSingleGpuInfer
11
  from diffusers.utils import export_to_video
12
  from diffusers.utils import load_image
13
  from PIL import Image
14
-
15
  import torch
16
 
17
  torch.backends.cuda.matmul.allow_tf32 = False
@@ -23,7 +23,7 @@ torch.backends.cudnn.benchmark = False
23
  torch.backends.cuda.preferred_blas_library="cublas"
24
  torch.backends.cuda.preferred_linalg_library="cusolver"
25
  torch.set_float32_matmul_precision("highest")
26
-
27
  os.putenv("HF_HUB_ENABLE_HF_TRANSFER","1")
28
  os.environ["SAFETENSORS_FAST_GPU"] = "1"
29
 
@@ -43,7 +43,7 @@ def init_predictor():
43
  )
44
  )
45
 
46
- @spaces.GPU(duration=90)
47
  def generate_video(prompt, image, size, steps, frames, guidance_scale, progress=gr.Progress(track_tqdm=True) ):
48
  print(f"image:{type(image)}")
49
  random.seed(time.time())
 
11
  from diffusers.utils import export_to_video
12
  from diffusers.utils import load_image
13
  from PIL import Image
14
+ '''
15
  import torch
16
 
17
  torch.backends.cuda.matmul.allow_tf32 = False
 
23
  torch.backends.cuda.preferred_blas_library="cublas"
24
  torch.backends.cuda.preferred_linalg_library="cusolver"
25
  torch.set_float32_matmul_precision("highest")
26
+ '''
27
  os.putenv("HF_HUB_ENABLE_HF_TRANSFER","1")
28
  os.environ["SAFETENSORS_FAST_GPU"] = "1"
29
 
 
43
  )
44
  )
45
 
46
+ @spaces.GPU(duration=120)
47
  def generate_video(prompt, image, size, steps, frames, guidance_scale, progress=gr.Progress(track_tqdm=True) ):
48
  print(f"image:{type(image)}")
49
  random.seed(time.time())