soonyau commited on
Commit
e9037d6
·
verified ·
1 Parent(s): 8b59682

Update cameractrlcmg/config.py

Browse files

disable all optimization to avoid set up issue

Files changed (1) hide show
  1. cameractrlcmg/config.py +3 -3
cameractrlcmg/config.py CHANGED
@@ -28,8 +28,8 @@ model = dict(
28
  space_scale=0.5,
29
  time_scale=1.0,
30
  from_pretrained=PRETRAINED_MODEL,
31
- enable_flashattn=True,
32
- enable_layernorm_kernel=True,
33
  camera_fuser_linear_dims=[MODEL_DIM, MODEL_DIM],
34
  camera_format=CAMERA_FORMAT
35
  )
@@ -42,7 +42,7 @@ text_encoder = dict(
42
  type="t5",
43
  from_pretrained="DeepFloyd/t5-v1_1-xxl",
44
  model_max_length=120,
45
- shardformer=True,
46
  cache_dir=CACHE_DIR,
47
  )
48
 
 
28
  space_scale=0.5,
29
  time_scale=1.0,
30
  from_pretrained=PRETRAINED_MODEL,
31
+ enable_flashattn=False,
32
+ enable_layernorm_kernel=False,
33
  camera_fuser_linear_dims=[MODEL_DIM, MODEL_DIM],
34
  camera_format=CAMERA_FORMAT
35
  )
 
42
  type="t5",
43
  from_pretrained="DeepFloyd/t5-v1_1-xxl",
44
  model_max_length=120,
45
+ shardformer=False,
46
  cache_dir=CACHE_DIR,
47
  )
48