anirudh-sub commited on
Commit
9dfa995
·
1 Parent(s): 685f928

Deleted tensor float in app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -3
app.py CHANGED
@@ -9,12 +9,9 @@ tokenizer = AutoTokenizer.from_pretrained(model)
9
  pipeline = transformers.pipeline(
10
  "text-generation",
11
  model=model,
12
- # torch_dtype=torch.float16,
13
  device_map="auto",
14
  )
15
 
16
- # torch.set_default_tensor_type(torch.FloatTensor)
17
-
18
  def debate_response(text):
19
  sequences = pipeline(
20
  "hello \n",
 
9
  pipeline = transformers.pipeline(
10
  "text-generation",
11
  model=model,
 
12
  device_map="auto",
13
  )
14
 
 
 
15
  def debate_response(text):
16
  sequences = pipeline(
17
  "hello \n",