block invalid RAS usages #22

Author: Teriks

dgenerate/pipelinewrapper/wrapper.py

@@ -3050,6 +3050,17 @@ def _auto_ras_check(self, args: DiffusionArguments):
                 raise _pipelines.UnsupportedPipelineConfigError(
                     'RAS does not support model CPU offloading.')
 
+            if args.ras_index_fusion and self.model_sequential_offload:
+                raise _pipelines.UnsupportedPipelineConfigError(
+                    'Index fusion is not supported for RAS when sequential offloading is enabled.')
+
+            if args.ras_index_fusion and (
+                    self.quantizer_uri or (self._unet_uri and _uris.UNetUri.parse(self._unet_uri).quantizer)
+            ):
+                raise _pipelines.UnsupportedPipelineConfigError(
+                    'Index fusion is not supported for RAS when UNet quantization is enabled, '
+                    'quantize the text encoders individually.')
+
             start_step = _types.default(args.ras_start_step, _constants.DEFAULT_RAS_START_STEP)
             end_step = _types.default(args.ras_end_step, args.inference_steps)
             if start_step > end_step:

dgenerate/renderloopconfig.py

@@ -1301,13 +1301,13 @@ def _check_help_arguments(self, a_namer: typing.Callable[[str], str]) -> bool:
 
         if text_encoder_help and len(self.text_encoder_uris) > 1:
             raise RenderLoopConfigError(
-                f'You cannot specify "help" or "helpargs" to {a_namer("text_encoders")} '
+                f'You cannot specify "help" or "helpargs" to {a_namer("text_encoder_uris")} '
                 f'with multiple values involved.'
             )
 
         if second_model_text_encoder_help and len(self.second_model_text_encoder_uris) > 1:
             raise RenderLoopConfigError(
-                f'You cannot specify "help" or "helpargs" to {a_namer("second_model_text_encoders")} '
+                f'You cannot specify "help" or "helpargs" to {a_namer("second_model_text_encoder_uris")} '
                 f'with multiple values involved.'
             )
 
@@ -1340,6 +1340,18 @@ def _check_optimization_features(self, a_namer: typing.Callable[[str], str]):
                 f'{a_namer("model_cpu_offload")} is not compatible '
                 f'with {a_namer("ras")} and related arguments.'
             )
+        if self.ras_index_fusion and self.model_sequential_offload:
+            raise RenderLoopConfigError(
+                f'{a_namer("ras_index_fusion")} is not compatible '
+                f'with {a_namer("model_sequential_offload")}.'
+            )
+
+        if self.ras_index_fusion and (
+                self.quantizer_uri or (self.unet_uri and _pipelinewrapper.UNetUri.parse(self.unet_uri).quantizer)
+        ):
+            raise RenderLoopConfigError(
+                f'{a_namer("ras_index_fusion")} is not supported for RAS when UNet quantization is enabled, '
+                f'quantize the text encoders individually using {a_namer("text_encoder_uris")}.')
 
     def _check_second_model_compatibility(self, a_namer: typing.Callable[[str], str]):
         """Check compatibility of second model arguments with the primary model type."""