Merge remote-tracking branch 'origin/ean-unify-sd' into ean-unify-sd

gpetters-amd · gpetters-amd · commit 0bfa2d98e713 · 2024-07-08T09:34:48.000-07:00
diff --git a/models/turbine_models/custom_models/llm_cmd_opts.py b/models/turbine_models/custom_models/llm_cmd_opts.py
@@ -44,8 +44,8 @@ def is_valid_file(arg):
 )
 
 ##############################################################################
-# SDXL Inference Options
-#    These options are used to control runtime parameters for SDXL inference.
+# Runtime Options
+#    These options are used to control runtime parameters.
 ##############################################################################
 
 p.add_argument(
diff --git a/models/turbine_models/custom_models/stateless_llama.py b/models/turbine_models/custom_models/stateless_llama.py
@@ -758,6 +758,9 @@ def format_out(results):
     llama = StatelessLlama(
         args.hf_model_name,
         args.precision,
+        args.max_length,
+        args.batch_size,
+        args.num_inference_steps,
         args.device,
         args.iree_target_triple,
         flags,

Original file line number	Diff line number	Diff line change
`@@ -44,8 +44,8 @@ def is_valid_file(arg):`
`44`	`44`	`)`
`45`	`45`
`46`	`46`	`##############################################################################`
`47`		`-# SDXL Inference Options`
`48`		`-# These options are used to control runtime parameters for SDXL inference.`
	`47`	`+# Runtime Options`
	`48`	`+# These options are used to control runtime parameters.`
`49`	`49`	`##############################################################################`
`50`	`50`
`51`	`51`	`p.add_argument(`