reshinthadith commited on
Commit
9ec1cea
1 Parent(s): 75cc022

Change the model from 13bit.

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,9 +13,9 @@ tokenizer = AutoTokenizer.from_pretrained(
13
  use_auth_token=auth_token if auth_token else True,
14
  )
15
  model = AutoModelForCausalLM.from_pretrained(
16
- "CarperAI/vicuna-13b-fine-tuned-rlhf-8bit",
17
  use_auth_token=auth_token if auth_token else True,
18
- )
19
 
20
 
21
  max_context_length = model.config.max_position_embeddings
 
13
  use_auth_token=auth_token if auth_token else True,
14
  )
15
  model = AutoModelForCausalLM.from_pretrained(
16
+ "CarperAI/vicuna-13b-fine-tuned-rlhf",
17
  use_auth_token=auth_token if auth_token else True,
18
+ ).to("cuda")
19
 
20
 
21
  max_context_length = model.config.max_position_embeddings