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)
|