1inkusFace commited on
Commit
face198
·
verified ·
1 Parent(s): 5b42e68

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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
 
@@ -80,7 +80,7 @@ with gr.Blocks() as demo:
80
  minimum=256,
81
  maximum=1024,
82
  step=16,
83
- value=448,
84
  )
85
  frames = gr.Slider(
86
  label="Number of Frames",
@@ -94,7 +94,7 @@ with gr.Blocks() as demo:
94
  minimum=1,
95
  maximum=96,
96
  step=1,
97
- value=20,
98
  )
99
  guidance_scale = gr.Slider(
100
  label="Guidance Scale",
 
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
 
 
80
  minimum=256,
81
  maximum=1024,
82
  step=16,
83
+ value=368,
84
  )
85
  frames = gr.Slider(
86
  label="Number of Frames",
 
94
  minimum=1,
95
  maximum=96,
96
  step=1,
97
+ value=25,
98
  )
99
  guidance_scale = gr.Slider(
100
  label="Guidance Scale",