Spaces:
Running on Zero
Running on Zero
Update pika_aoti.py
Browse files- 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", "
|
| 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"}
|