Skip to content

Commit 0bfa2d9

Browse files
committed
Merge remote-tracking branch 'origin/ean-unify-sd' into ean-unify-sd
2 parents 90844af + 16ee249 commit 0bfa2d9

File tree

2 files changed

+5
-2
lines changed

2 files changed

+5
-2
lines changed

models/turbine_models/custom_models/llm_cmd_opts.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -44,8 +44,8 @@ def is_valid_file(arg):
4444
)
4545

4646
##############################################################################
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.
4949
##############################################################################
5050

5151
p.add_argument(

models/turbine_models/custom_models/stateless_llama.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -758,6 +758,9 @@ def format_out(results):
758758
llama = StatelessLlama(
759759
args.hf_model_name,
760760
args.precision,
761+
args.max_length,
762+
args.batch_size,
763+
args.num_inference_steps,
761764
args.device,
762765
args.iree_target_triple,
763766
flags,

0 commit comments

Comments
 (0)