arjunanand13 commited on
Commit
106a698
1 Parent(s): d145f19

Update process.py

Browse files
Files changed (1) hide show
  1. process.py +1 -1
process.py CHANGED
@@ -12,7 +12,7 @@ DEVICE = 'cuda' if torch.cuda.is_available() else 'cpu'
12
  TORCH_TYPE = torch.bfloat16 if torch.cuda.is_available() and torch.cuda.get_device_capability()[
13
  0] >= 8 else torch.float16
14
 
15
- parser = argparse.ArgumentParser(description="CogVLM2-Video CLI Demo")
16
  parser.add_argument('--quant', type=int, choices=[4, 8], help='Enable 4-bit or 8-bit precision loading', default=4)
17
  args = parser.parse_args([])
18
 
 
12
  TORCH_TYPE = torch.bfloat16 if torch.cuda.is_available() and torch.cuda.get_device_capability()[
13
  0] >= 8 else torch.float16
14
 
15
+ parser = argparse.ArgumentParser(description="Apollo tyre delay reasoning")
16
  parser.add_argument('--quant', type=int, choices=[4, 8], help='Enable 4-bit or 8-bit precision loading', default=4)
17
  args = parser.parse_args([])
18