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

Changed the model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -4,12 +4,12 @@ from transformers import AutoTokenizer
4
  import torch
5
  from diffusers.utils.torch_utils import randn_tensor
6
 
7
- model = "anirudh-sub/debate_model_v2.1"
8
  tokenizer = AutoTokenizer.from_pretrained(model)
9
  pipeline = transformers.pipeline(
10
  "text-generation",
11
  model=model,
12
- torch_dtype=torch.float32,
13
  device_map="auto",
14
  )
15
 
 
4
  import torch
5
  from diffusers.utils.torch_utils import randn_tensor
6
 
7
+ model = "Glasshes/debate_v3.1_model"
8
  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