Spaces:

Staticaliza
/

Sense

Running

Staticaliza commited on May 28

Commit

5e7d02a

verified ·

1 Parent(s): c0d6cf5

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,7 @@ DEFAULT_INPUT = "Describe in one paragraph."
 MAX_FRAMES = 64
 repo_name = "openbmb/MiniCPM-o-2_6" # "openbmb/MiniCPM-V-2_6-int4" # "openbmb/MiniCPM-V-2_6"
-repo = AutoModel.from_pretrained(repo_name', trust_remote_code=True, attn_implementation='sdpa', torch_dtype=torch.bfloat16)
 tokenizer = AutoTokenizer.from_pretrained(repo_name, trust_remote_code=True)
 css = '''

 MAX_FRAMES = 64
 repo_name = "openbmb/MiniCPM-o-2_6" # "openbmb/MiniCPM-V-2_6-int4" # "openbmb/MiniCPM-V-2_6"
+repo = AutoModel.from_pretrained(repo_name, trust_remote_code=True, attn_implementation="sdpa", torch_dtype=torch.bfloat16)
 tokenizer = AutoTokenizer.from_pretrained(repo_name, trust_remote_code=True)
 css = '''