LPX55 commited on
Commit
f7dbffd
·
verified ·
1 Parent(s): 8849a43

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ feature_extractor = SiglipImageProcessor.from_pretrained("lllyasviel/flux_redux_
44
  image_encoder = SiglipVisionModel.from_pretrained("lllyasviel/flux_redux_bfl", subfolder='image_encoder', torch_dtype=torch.float16).cpu()
45
 
46
  # quant_config = DiffusersBitsAndBytesConfig(load_in_8bit=True)
47
- transformer = HunyuanVideoTransformer3DModelPacked.from_pretrained('sirolim/FramePack_F1_I2V_FP8', torch_dtype=torch.bfloat16).cpu()
48
  # transformer = HunyuanVideoTransformer3DModelPacked.from_pretrained('sirolim/FramePack_F1_I2V_FP8', torch_dtype=torch.bfloat16).cpu()
49
 
50
  vae.eval()
 
44
  image_encoder = SiglipVisionModel.from_pretrained("lllyasviel/flux_redux_bfl", subfolder='image_encoder', torch_dtype=torch.float16).cpu()
45
 
46
  # quant_config = DiffusersBitsAndBytesConfig(load_in_8bit=True)
47
+ transformer = HunyuanVideoTransformer3DModelPacked.from_pretrained('sirolim/FramePack_F1_I2V_FP8', "FramePack_F1_I2V_HY_fp8_e4m3fn.safetensors", use_safetensors=True, torch_dtype=torch.bfloat16).cpu()
48
  # transformer = HunyuanVideoTransformer3DModelPacked.from_pretrained('sirolim/FramePack_F1_I2V_FP8', torch_dtype=torch.bfloat16).cpu()
49
 
50
  vae.eval()