weiyuchoumou526 commited on
Commit
c691e3c
·
1 Parent(s): 57c8fde

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -43,9 +43,6 @@ transformer_path = "Kunbyte/ROSE"
43
  config_path = "./configs/wan2.1/wan_civitai.yaml"
44
  config = OmegaConf.load(config_path)
45
 
46
- tokenizer_subpath = config['text_encoder_kwargs'].get('tokenizer_subpath', 'tokenizer')
47
- tokenizer = AutoTokenizer.from_pretrained(pretrained_model_path, tokenizer_subpath)
48
-
49
  text_encoder_subpath = config['text_encoder_kwargs'].get('text_encoder_subpath', 'text_encoder')
50
  text_encoder = WanT5EncoderModel.from_pretrained(
51
  pretrained_model_path,
@@ -75,6 +72,8 @@ noise_scheduler = FlowMatchEulerDiscreteScheduler(
75
  **filter_kwargs(FlowMatchEulerDiscreteScheduler, OmegaConf.to_container(config['scheduler_kwargs']))
76
  )
77
 
 
 
78
  # tokenizer = AutoTokenizer.from_pretrained(
79
  # os.path.join(pretrained_model_path, config['text_encoder_kwargs'].get('tokenizer_subpath', 'tokenizer')),
80
  # )
 
43
  config_path = "./configs/wan2.1/wan_civitai.yaml"
44
  config = OmegaConf.load(config_path)
45
 
 
 
 
46
  text_encoder_subpath = config['text_encoder_kwargs'].get('text_encoder_subpath', 'text_encoder')
47
  text_encoder = WanT5EncoderModel.from_pretrained(
48
  pretrained_model_path,
 
72
  **filter_kwargs(FlowMatchEulerDiscreteScheduler, OmegaConf.to_container(config['scheduler_kwargs']))
73
  )
74
 
75
+ tokenizer_subpath = config['text_encoder_kwargs'].get('tokenizer_subpath', 'tokenizer')
76
+ tokenizer = AutoTokenizer.from_pretrained(pretrained_model_path, tokenizer_subpath)
77
  # tokenizer = AutoTokenizer.from_pretrained(
78
  # os.path.join(pretrained_model_path, config['text_encoder_kwargs'].get('tokenizer_subpath', 'tokenizer')),
79
  # )