code: def test_pipeline2():
      SEED = 1236
      set_seed(SEED)
  
      import torch
      from src.h2oai_pipeline import H2OTextGenerationPipeline
      from transformers import AutoModelForCausalLM, AutoTokenizer
  
      model_name = "h2oai/h2ogpt-oig-oasst1-512-6_9b"
      load_in_8bit = False
      device_map = {"": 0}
  
      tokenizer = AutoTokenizer.from_pretrained(
          model_name, padding_side="left"
      )
      model = AutoModelForCausalLM.from_pretrained(
          model_name,
          torch_dtype=torch.bfloat16,
          device_map=device_map,
          load_in_8bit=load_in_8bit,
      )
      generate_text = H2OTextGenerationPipeline(
          model=model,
          tokenizer=tokenizer,
          prompt_type="human_bot",
          base_model=model_name,
      )
  
      res = generate_text(
          "Why is drinking water so healthy?", max_new_tokens=100
      )
      print(res[0]["generated_text"])
  
      assert (
          "Drinking water is so healthy because it is full of nutrients and other beneficial substances"
          in res[0]["generated_text"]
      )
  
imports: import pytest
  from tests.utils import wrap_test_forked
  from src.utils import set_seed
label_desc: Use image node_id 3 for calling a global function with example usage: test_pipeline2() without return types