Update app.py

app.py

@@ -56,6 +56,7 @@ from diffusers import (
 )
 import evaluate as hf_evaluate
 from jinja2 import Template
+import spaces
 
 logger = logging.getLogger(__name__)
 
@@ -549,6 +550,7 @@ def _create_training_args(output_dir, repo_id, **kwargs):
         "report_to": "wandb" if kwargs.get('wandb_api_key_input') else "none",
         "remove_unused_columns": False,
         "group_by_length": kwargs.get('group_by_length', False),
+        "packing": kwargs.get('packing', False),
         "metric_for_best_model": kwargs.get('metric_for_best_model', 'loss') if kwargs.get('run_evaluation') else None,
         "greater_is_better": kwargs.get('greater_is_better', False),
         "neftune_noise_alpha": neftune_alpha if neftune_alpha > 0 else None,