adhvaithprasad commited on
Commit
288df40
1 Parent(s): 4ac374e

imported torch

Browse files
Files changed (1) hide show
  1. customerSupport.py +1 -0
customerSupport.py CHANGED
@@ -2,6 +2,7 @@
2
  from transformers import AutoModelForCausalLM,GenerationConfig
3
  from peft import AutoPeftModelForCausalLM
4
  from peft import PeftModel, PeftConfig
 
5
 
6
  def input_data_preprocessing(example):
7
 
 
2
  from transformers import AutoModelForCausalLM,GenerationConfig
3
  from peft import AutoPeftModelForCausalLM
4
  from peft import PeftModel, PeftConfig
5
+ import torch
6
 
7
  def input_data_preprocessing(example):
8