Spaces:
Runtime error
Runtime error
Update process.py
Browse files- 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="
|
| 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 |
|