Fabrice-TIERCELIN commited on
Commit
511b862
·
verified ·
1 Parent(s): 46eaf47

Change model

Browse files
Files changed (1) hide show
  1. app_endframe.py +1 -1
app_endframe.py CHANGED
@@ -64,7 +64,7 @@ vae = AutoencoderKLHunyuanVideo.from_pretrained("Fabrice-TIERCELIN/HunyuanVideo"
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()
66
 
67
- transformer = HunyuanVideoTransformer3DModelPacked.from_pretrained('lllyasviel/FramePackI2V_HY', torch_dtype=torch.bfloat16).cpu()
68
 
69
  vae.eval()
70
  text_encoder.eval()
 
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()
66
 
67
+ transformer = HunyuanVideoTransformer3DModelPacked.from_pretrained('Fabrice-TIERCELIN/FramePackI2V_HY', torch_dtype=torch.bfloat16).cpu()
68
 
69
  vae.eval()
70
  text_encoder.eval()