JiantaoLin commited on
Commit
b67d0e7
·
1 Parent(s): 1809f2a
Files changed (1) hide show
  1. pipeline/kiss3d_wrapper.py +1 -1
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