chunk_content: ataLoader
  from tqdm import tqdm
  
  device = "cuda"
  model_name_or_path = "bigscience/bloomz-560m"
  tokenizer_name_or_path = "bigscience/bloomz-560m"
  peft_config = PromptTuningConfig(
      task_type=TaskType.CAUSAL_LM,
      prompt_tuning_init=PromptTuningInit.TEXT,
      num_virtual_tokens=8,
      prompt_tuning_init_text="Classify if the tweet is a complaint or not:",
      tokenizer_name_or_path=model_name_or_path,
  )
  
  dataset_name = "twitter_complaints"
  c