xzerus commited on
Commit
fcc0cc5
·
verified ·
1 Parent(s): 42699b5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ model = AutoModel.from_pretrained(
10
  low_cpu_mem_usage=True,
11
  trust_remote_code=True,
12
  use_flash_attn=False # Disable Flash Attention
13
- ).eval() # Do not move to CUDA, leave on CPU
14
 
15
  # Load the image processor
16
  image_processor = CLIPImageProcessor.from_pretrained('OpenGVLab/InternVL2_5-1B')
 
10
  low_cpu_mem_usage=True,
11
  trust_remote_code=True,
12
  use_flash_attn=False # Disable Flash Attention
13
+ ).eval() # Do not move to CUDA, force CPU execution
14
 
15
  # Load the image processor
16
  image_processor = CLIPImageProcessor.from_pretrained('OpenGVLab/InternVL2_5-1B')