Hugging Face's logo

Spaces:
jiuface
/
florence-sam-masking
Running on Zero

App Files Community
jiuface commited on
Commit
9947239
•
1 Parent(s): ffe5aff

bugfix

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ import boto3
23
  from datetime import datetime
24
 
25
  DEVICE = torch.device("cuda")
26
- # DEVICE = torch.device("cpu")
27
 
28
  torch.autocast(device_type="cuda", dtype=torch.bfloat16).__enter__()
29
  if torch.cuda.get_device_properties(0).major >= 8:
 
23
  from datetime import datetime
24
 
25
  DEVICE = torch.device("cuda")
26
+ MAX_SEED = np.iinfo(np.int32).max
27
 
28
  torch.autocast(device_type="cuda", dtype=torch.bfloat16).__enter__()
29
  if torch.cuda.get_device_properties(0).major >= 8: