Spaces:
Runtime error
Runtime error
https://huggingface.co/Fabrice-TIERCELIN/HunyuanVideo
Browse files- app_endframe.py +5 -5
app_endframe.py
CHANGED
|
@@ -55,11 +55,11 @@ high_vram = free_mem_gb > 60
|
|
| 55 |
print(f'Free VRAM {free_mem_gb} GB')
|
| 56 |
print(f'High-VRAM Mode: {high_vram}')
|
| 57 |
|
| 58 |
-
text_encoder = LlamaModel.from_pretrained("
|
| 59 |
-
text_encoder_2 = CLIPTextModel.from_pretrained("
|
| 60 |
-
tokenizer = LlamaTokenizerFast.from_pretrained("
|
| 61 |
-
tokenizer_2 = CLIPTokenizer.from_pretrained("
|
| 62 |
-
vae = AutoencoderKLHunyuanVideo.from_pretrained("
|
| 63 |
|
| 64 |
feature_extractor = SiglipImageProcessor.from_pretrained("lllyasviel/flux_redux_bfl", subfolder='feature_extractor')
|
| 65 |
image_encoder = SiglipVisionModel.from_pretrained("lllyasviel/flux_redux_bfl", subfolder='image_encoder', torch_dtype=torch.float16).cpu()
|
|
|
|
| 55 |
print(f'Free VRAM {free_mem_gb} GB')
|
| 56 |
print(f'High-VRAM Mode: {high_vram}')
|
| 57 |
|
| 58 |
+
text_encoder = LlamaModel.from_pretrained("Fabrice-TIERCELIN/HunyuanVideo", subfolder='text_encoder', torch_dtype=torch.float16).cpu()
|
| 59 |
+
text_encoder_2 = CLIPTextModel.from_pretrained("Fabrice-TIERCELIN/HunyuanVideo", subfolder='text_encoder_2', torch_dtype=torch.float16).cpu()
|
| 60 |
+
tokenizer = LlamaTokenizerFast.from_pretrained("Fabrice-TIERCELIN/HunyuanVideo", subfolder='tokenizer')
|
| 61 |
+
tokenizer_2 = CLIPTokenizer.from_pretrained("Fabrice-TIERCELIN/HunyuanVideo", subfolder='tokenizer_2')
|
| 62 |
+
vae = AutoencoderKLHunyuanVideo.from_pretrained("Fabrice-TIERCELIN/HunyuanVideo", subfolder='vae', torch_dtype=torch.float16).cpu()
|
| 63 |
|
| 64 |
feature_extractor = SiglipImageProcessor.from_pretrained("lllyasviel/flux_redux_bfl", subfolder='feature_extractor')
|
| 65 |
image_encoder = SiglipVisionModel.from_pretrained("lllyasviel/flux_redux_bfl", subfolder='image_encoder', torch_dtype=torch.float16).cpu()
|