Daemontatox commited on
Commit
c521b49
·
verified ·
1 Parent(s): c6ee6e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -17,7 +17,7 @@ logging.basicConfig(level=logging.INFO)
17
  logger = logging.getLogger(__name__)
18
 
19
  # Load model and processor
20
- ckpt = "Daemontatox/DocumentCogito"
21
  model = MllamaForConditionalGeneration.from_pretrained(ckpt, torch_dtype=torch.bfloat16).to("cuda")
22
  processor = AutoProcessor.from_pretrained(ckpt)
23
 
 
17
  logger = logging.getLogger(__name__)
18
 
19
  # Load model and processor
20
+ ckpt = "deepseek-ai/deepseek-vl2"
21
  model = MllamaForConditionalGeneration.from_pretrained(ckpt, torch_dtype=torch.bfloat16).to("cuda")
22
  processor = AutoProcessor.from_pretrained(ckpt)
23