Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -21,21 +21,24 @@ torch.backends.cudnn.benchmark = True
|
|
21 |
#torch.backends.cuda.preferred_linalg_library="cusolver"
|
22 |
torch.set_float32_matmul_precision("highest")
|
23 |
os.putenv("HF_HUB_ENABLE_HF_TRANSFER","1")
|
24 |
-
HF_TOKEN = os.getenv("HF_TOKEN")
|
25 |
os.environ["SAFETENSORS_FAST_GPU"] = "1"
|
26 |
MAX_SEED = np.iinfo(np.int64).max
|
27 |
|
28 |
-
single_file_url = "https://huggingface.co/Lightricks/LTX-Video/ltx-video-2b-v0.9.1.safetensors"
|
|
|
29 |
|
30 |
pipe = LTXImageToVideoPipeline.from_pretrained(
|
31 |
"Lightricks/LTX-Video",
|
32 |
-
token=HF_TOKEN,
|
33 |
transformer=None,
|
34 |
text_encoder=None,
|
35 |
).to(torch.device("cuda"),torch.bfloat16)
|
36 |
|
37 |
text_encoder = T5EncoderModel.from_pretrained("Lightricks/LTX-Video",subfolder='text_encoder',token=True).to(torch.device("cuda"),torch.bfloat16)
|
38 |
-
transformer = LTXVideoTransformer3DModel.from_single_file(single_file_url,
|
|
|
|
|
39 |
|
40 |
@spaces.GPU(duration=80)
|
41 |
def generate_video(
|
@@ -93,4 +96,4 @@ iface = gr.Interface(
|
|
93 |
description="Generate video from image with LTX-Image-to-Video.",
|
94 |
)
|
95 |
|
96 |
-
iface.launch()
|
|
|
21 |
#torch.backends.cuda.preferred_linalg_library="cusolver"
|
22 |
torch.set_float32_matmul_precision("highest")
|
23 |
os.putenv("HF_HUB_ENABLE_HF_TRANSFER","1")
|
24 |
+
#HF_TOKEN = os.getenv("HF_TOKEN")
|
25 |
os.environ["SAFETENSORS_FAST_GPU"] = "1"
|
26 |
MAX_SEED = np.iinfo(np.int64).max
|
27 |
|
28 |
+
#single_file_url = "https://huggingface.co/Lightricks/LTX-Video/ltx-video-2b-v0.9.1.safetensors"
|
29 |
+
single_file_url = "https://huggingface.co/Lightricks/LTX-Video/ltx-video-2b-v0.9.5.safetensors"
|
30 |
|
31 |
pipe = LTXImageToVideoPipeline.from_pretrained(
|
32 |
"Lightricks/LTX-Video",
|
33 |
+
#token=HF_TOKEN,
|
34 |
transformer=None,
|
35 |
text_encoder=None,
|
36 |
).to(torch.device("cuda"),torch.bfloat16)
|
37 |
|
38 |
text_encoder = T5EncoderModel.from_pretrained("Lightricks/LTX-Video",subfolder='text_encoder',token=True).to(torch.device("cuda"),torch.bfloat16)
|
39 |
+
transformer = LTXVideoTransformer3DModel.from_single_file(single_file_url,
|
40 |
+
#token=HF_TOKEN
|
41 |
+
).to(torch.device("cuda"),torch.bfloat16)
|
42 |
|
43 |
@spaces.GPU(duration=80)
|
44 |
def generate_video(
|
|
|
96 |
description="Generate video from image with LTX-Image-to-Video.",
|
97 |
)
|
98 |
|
99 |
+
iface.launch(share = True)
|