Hugging Face's logo

Spaces:
CCCCCC
/
BPO_demo
Runtime error

App Files Community
1
CCCCCC commited on
Commit
892aa5d
·
1 Parent(s): e232fd5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ import torch
8
 
9
  model_path = 'THUDM/BPO'
10
 
11
- device = 'cuda'
12
 
13
  tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True, add_prefix_space=True)
14
  model = AutoModelForCausalLM.from_pretrained(model_path, trust_remote_code=True, device_map=device, load_in_8bit=True)
 
8
 
9
  model_path = 'THUDM/BPO'
10
 
11
+ device = 'cuda:0'
12
 
13
  tokenizer = AutoTokenizer.from_pretrained(model_path, trust_remote_code=True, add_prefix_space=True)
14
  model = AutoModelForCausalLM.from_pretrained(model_path, trust_remote_code=True, device_map=device, load_in_8bit=True)