anvilarth commited on
Commit
9c51397
·
verified ·
1 Parent(s): 2287fea

Update Garage/AugmenterPipeline.py

Browse files
Files changed (1) hide show
  1. Garage/AugmenterPipeline.py +1 -0
Garage/AugmenterPipeline.py CHANGED
@@ -1,6 +1,7 @@
1
  import torch
2
  import numpy as np
3
  import random
 
4
  from PIL import Image
5
  from typing import List, Optional, Tuple
6
  from accelerate import Accelerator
 
1
  import torch
2
  import numpy as np
3
  import random
4
+ import spaces
5
  from PIL import Image
6
  from typing import List, Optional, Tuple
7
  from accelerate import Accelerator