JiantaoLin
commited on
Commit
·
b67d0e7
1
Parent(s):
1809f2a
new
Browse files
pipeline/kiss3d_wrapper.py
CHANGED
@@ -89,7 +89,7 @@ def init_wrapper_from_config(config_path):
|
|
89 |
# load redux model
|
90 |
flux_redux_pipe = None
|
91 |
if flux_redux_pth is not None:
|
92 |
-
flux_redux_pipe = FluxPriorReduxPipeline.from_pretrained(flux_redux_pth, torch_dtype=torch.bfloat16)
|
93 |
flux_redux_pipe.text_encoder = flux_pipe.text_encoder
|
94 |
flux_redux_pipe.text_encoder_2 = flux_pipe.text_encoder_2
|
95 |
flux_redux_pipe.tokenizer = flux_pipe.tokenizer
|
|
|
89 |
# load redux model
|
90 |
flux_redux_pipe = None
|
91 |
if flux_redux_pth is not None:
|
92 |
+
flux_redux_pipe = FluxPriorReduxPipeline.from_pretrained(flux_redux_pth, torch_dtype=torch.bfloat16, token=access_token)
|
93 |
flux_redux_pipe.text_encoder = flux_pipe.text_encoder
|
94 |
flux_redux_pipe.text_encoder_2 = flux_pipe.text_encoder_2
|
95 |
flux_redux_pipe.tokenizer = flux_pipe.tokenizer
|