anirudh-sub commited on
Commit
b7a5805
·
1 Parent(s): a54fee0

Testing tensor.float32 with app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -9,7 +9,7 @@ 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
 
@@ -23,7 +23,6 @@ def debate_response(text):
23
  eos_token_id=tokenizer.eos_token_id,
24
  max_length=100,
25
  )
26
- print(sequences)
27
  return "testing"
28
 
29
 
 
9
  pipeline = transformers.pipeline(
10
  "text-generation",
11
  model=model,
12
+ torch_dtype=torch.float32,
13
  device_map="auto",
14
  )
15
 
 
23
  eos_token_id=tokenizer.eos_token_id,
24
  max_length=100,
25
  )
 
26
  return "testing"
27
 
28