chenlin commited on
Commit
15f5821
1 Parent(s): 4dcb992
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -44,7 +44,7 @@ The service is a research preview intended for non-commercial use only, subject
44
  new_path = 'Lin-Chen/ShareCaptioner-Video'
45
  tokenizer = AutoTokenizer.from_pretrained(new_path, trust_remote_code=True)
46
  model = AutoModel.from_pretrained(
47
- new_path, torch_dtype=torch.bfloat16, trust_remote_code=True).cuda().eval()
48
  model.cuda()
49
  model.tokenizer = tokenizer
50
 
 
44
  new_path = 'Lin-Chen/ShareCaptioner-Video'
45
  tokenizer = AutoTokenizer.from_pretrained(new_path, trust_remote_code=True)
46
  model = AutoModel.from_pretrained(
47
+ new_path, torch_dtype=torch.float16, trust_remote_code=True).cuda().eval()
48
  model.cuda()
49
  model.tokenizer = tokenizer
50