Update app.py

app.py

@@ -6,10 +6,8 @@ import os
 
 #CHECKPOINT_PATH=f'output_lh/checkpoint-600'
 tokenizer = AutoTokenizer.from_pretrained("chatglm3-6b", trust_remote_code=True)
-config = AutoConfig.from_pretrained("chatglm3-6b", trust_remote_code=True, pre_seq_len=128)
-model = AutoModel.from_pretrained("chatglm3-6b", config=config, trust_remote_code=True)
 
-model = model.half().float()
+model = AutoModel.from_pretrained("THUDM/chatglm3-6b", trust_remote_code=True, device='cuda')
 
 model = model.eval()