Staticaliza commited on
Commit
cfd4c18
1 Parent(s): 029aec2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -17,8 +17,9 @@ print(f"[SYSTEM] | Using {DEVICE} type compute device.")
17
  DEFAULT_INPUT = "Describe in one paragraph."
18
  MAX_FRAMES = 64
19
 
20
- repo = AutoModel.from_pretrained("openbmb/MiniCPM-V-2_6", torch_dtype=torch.bfloat16, trust_remote_code=True)
21
- tokenizer = AutoTokenizer.from_pretrained("openbmb/MiniCPM-V-2_6", trust_remote_code=True)
 
22
 
23
  css = '''
24
  .gradio-container{max-width: 560px !important}
 
17
  DEFAULT_INPUT = "Describe in one paragraph."
18
  MAX_FRAMES = 64
19
 
20
+ repo_name = "openbmb/MiniCPM-V-2_6-int4" # "openbmb/MiniCPM-V-2_6"
21
+ repo = AutoModel.from_pretrained(repo_name, trust_remote_code=True)
22
+ tokenizer = AutoTokenizer.from_pretrained(repo_name, trust_remote_code=True)
23
 
24
  css = '''
25
  .gradio-container{max-width: 560px !important}