Removed has_text_input
Browse files
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,
|