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
|