Spaces:
Sleeping
Sleeping
arjunanand13
commited on
Commit
•
106a698
1
Parent(s):
d145f19
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 |
|