Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
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()
|