sdxl prompt encoder typo (#778)

nod-ai · Jul 16, 2024 · d24df8f · d24df8f
1 parent 7570e69
commit d24df8f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models/turbine_models/custom_models/sdxl_inference/sdxl_prompt_encoder.py b/models/turbine_models/custom_models/sdxl_inference/sdxl_prompt_encoder.py
@@ -220,7 +220,7 @@ def export_prompt_encoder(
 
     input_batchsize = 1
     if batch_input:
-        input_batchsize = batchsize
+        input_batchsize = batch_size
 
     if precision == "fp16":
         prompt_encoder_module = prompt_encoder_module.half()