Diskfo commited on
Commit
e003886
路
verified 路
1 Parent(s): 3436cfb

Update pika_aoti.py

Browse files
Files changed (1) hide show
  1. pika_aoti.py +1 -1
pika_aoti.py CHANGED
@@ -8,7 +8,7 @@ import torch
8
  BLOCK_CONTAINER = "transformer_blocks"
9
  TEMB_ROWS = int(os.environ.get("PIKA_AOTI_TEMB_ROWS", "4"))
10
  AOTI = os.environ.get("PIKA_AOTI", "0") == "1"
11
- AOTI_REPO = os.environ.get("PIKA_AOTI_REPO", os.environ.get("PIKA_MODEL_REPO", "diffusers-internal-dev/Pika-H3-Flex"))
12
  AOTI_REVISION = os.environ.get("PIKA_AOTI_REVISION")
13
  INDUCTOR_CONFIGS = {"max_autotune": True, "coordinate_descent_tuning": True, "triton.cudagraphs": False}
14
  STATUS = {"state": "off"}
 
8
  BLOCK_CONTAINER = "transformer_blocks"
9
  TEMB_ROWS = int(os.environ.get("PIKA_AOTI_TEMB_ROWS", "4"))
10
  AOTI = os.environ.get("PIKA_AOTI", "0") == "1"
11
+ AOTI_REPO = os.environ.get("PIKA_AOTI_REPO", os.environ.get("PIKA_MODEL_REPO", "Pika-Labs/Pika-H3-Flex"))
12
  AOTI_REVISION = os.environ.get("PIKA_AOTI_REVISION")
13
  INDUCTOR_CONFIGS = {"max_autotune": True, "coordinate_descent_tuning": True, "triton.cudagraphs": False}
14
  STATUS = {"state": "off"}