Staticaliza commited on
Commit
e52a62d
·
verified ·
1 Parent(s): 1c5a322

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ print(f"[SYSTEM] | Using {DEVICE} type compute device.")
18
  DEFAULT_INPUT = "Describe in one paragraph."
19
  MAX_FRAMES = 64
20
 
21
- model_name = "2dameneko/MiniCPM-o-2_6-nf4"
22
 
23
  repo = AutoModel.from_pretrained(model_name, trust_remote_code=True, attn_implementation="sdpa", torch_dtype=torch.bfloat16).to(DEVICE)
24
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
 
18
  DEFAULT_INPUT = "Describe in one paragraph."
19
  MAX_FRAMES = 64
20
 
21
+ model_name = "openbmb/MiniCPM-o-2_6"
22
 
23
  repo = AutoModel.from_pretrained(model_name, trust_remote_code=True, attn_implementation="sdpa", torch_dtype=torch.bfloat16).to(DEVICE)
24
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)