Spaces:
Paused
Paused
Update app.py
Browse files
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=
|
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())
|