manu commited on
Commit
9c9e0a5
β€’
1 Parent(s): 0d01d71

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -68,7 +68,7 @@ model = ColPali.from_pretrained(
68
  "google/paligemma-3b-mix-448", torch_dtype=torch.bfloat16, device_map="cuda", token = token).eval()
69
 
70
  model.load_adapter(model_name)
71
- processor = AutoProcessor.from_pretrained(model_name, cache_dir=cache_dir, token = token)
72
 
73
  device = model.device
74
 
 
68
  "google/paligemma-3b-mix-448", torch_dtype=torch.bfloat16, device_map="cuda", token = token).eval()
69
 
70
  model.load_adapter(model_name)
71
+ processor = AutoProcessor.from_pretrained(model_name, token = token)
72
 
73
  device = model.device
74