Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -25,8 +25,8 @@ os.putenv("HF_HUB_ENABLE_HF_TRANSFER","1")
|
|
| 25 |
os.environ["SAFETENSORS_FAST_GPU"] = "1"
|
| 26 |
MAX_SEED = np.iinfo(np.int64).max
|
| 27 |
|
| 28 |
-
|
| 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",
|
|
@@ -35,7 +35,9 @@ pipe = LTXImageToVideoPipeline.from_pretrained(
|
|
| 35 |
text_encoder=None,
|
| 36 |
).to(torch.device("cuda"),torch.bfloat16)
|
| 37 |
|
| 38 |
-
text_encoder = T5EncoderModel.from_pretrained("Lightricks/LTX-Video",subfolder='text_encoder',
|
|
|
|
|
|
|
| 39 |
transformer = LTXVideoTransformer3DModel.from_single_file(single_file_url,
|
| 40 |
#token=HF_TOKEN
|
| 41 |
).to(torch.device("cuda"),torch.bfloat16)
|
|
|
|
| 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",
|
|
|
|
| 35 |
text_encoder=None,
|
| 36 |
).to(torch.device("cuda"),torch.bfloat16)
|
| 37 |
|
| 38 |
+
text_encoder = T5EncoderModel.from_pretrained("Lightricks/LTX-Video",subfolder='text_encoder',
|
| 39 |
+
#token=True
|
| 40 |
+
).to(torch.device("cuda"),torch.bfloat16)
|
| 41 |
transformer = LTXVideoTransformer3DModel.from_single_file(single_file_url,
|
| 42 |
#token=HF_TOKEN
|
| 43 |
).to(torch.device("cuda"),torch.bfloat16)
|