sheldonl commited on
Commit
1cc7f6f
·
1 Parent(s): 07005c9

Removed has_text_input

Browse files
Files changed (1) hide show
  1. world_generation_pipeline.py +4 -4
world_generation_pipeline.py CHANGED
@@ -164,7 +164,7 @@ class ARBaseGenerationPipeline(BaseWorldGenerationPipeline):
164
  inference_type: str,
165
  checkpoint_dir: str,
166
  checkpoint_name: str,
167
- has_text_input: bool = False,
168
  offload_network: bool = False,
169
  offload_tokenizer: bool = False,
170
  disable_diffusion_decoder: bool = False,
@@ -234,7 +234,7 @@ class ARBaseGenerationPipeline(BaseWorldGenerationPipeline):
234
  inference_type=inference_type,
235
  checkpoint_dir=checkpoint_dir,
236
  checkpoint_name=checkpoint_name,
237
- has_text_input=has_text_input,
238
  offload_guardrail_models=offload_guardrail_models,
239
  offload_network=offload_network,
240
  offload_tokenizer=offload_tokenizer,
@@ -680,7 +680,7 @@ class ARVideo2WorldGenerationPipeline(ARBaseGenerationPipeline):
680
  checkpoint_dir: str,
681
  checkpoint_name: str,
682
  inference_type: str = None,
683
- has_text_input: bool = True,
684
  disable_diffusion_decoder: bool = False,
685
  offload_guardrail_models: bool = False,
686
  offload_diffusion_decoder: bool = False,
@@ -706,7 +706,7 @@ class ARVideo2WorldGenerationPipeline(ARBaseGenerationPipeline):
706
  checkpoint_dir=checkpoint_dir,
707
  checkpoint_name=checkpoint_name,
708
  inference_type=inference_type,
709
- has_text_input=has_text_input,
710
  disable_diffusion_decoder=disable_diffusion_decoder,
711
  offload_guardrail_models=offload_guardrail_models,
712
  offload_diffusion_decoder=offload_diffusion_decoder,
 
164
  inference_type: str,
165
  checkpoint_dir: str,
166
  checkpoint_name: str,
167
+ #has_text_input: bool = False,
168
  offload_network: bool = False,
169
  offload_tokenizer: bool = False,
170
  disable_diffusion_decoder: bool = False,
 
234
  inference_type=inference_type,
235
  checkpoint_dir=checkpoint_dir,
236
  checkpoint_name=checkpoint_name,
237
+ #has_text_input=has_text_input,
238
  offload_guardrail_models=offload_guardrail_models,
239
  offload_network=offload_network,
240
  offload_tokenizer=offload_tokenizer,
 
680
  checkpoint_dir: str,
681
  checkpoint_name: str,
682
  inference_type: str = None,
683
+ #has_text_input: bool = True,
684
  disable_diffusion_decoder: bool = False,
685
  offload_guardrail_models: bool = False,
686
  offload_diffusion_decoder: bool = False,
 
706
  checkpoint_dir=checkpoint_dir,
707
  checkpoint_name=checkpoint_name,
708
  inference_type=inference_type,
709
+ #has_text_input=has_text_input,
710
  disable_diffusion_decoder=disable_diffusion_decoder,
711
  offload_guardrail_models=offload_guardrail_models,
712
  offload_diffusion_decoder=offload_diffusion_decoder,