korbip commited on
Commit
1b9ccd8
·
verified ·
1 Parent(s): 51088a3

Point the fla fork at OpenEuroLLM/ComplexKDA

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. modeling_complex_kda.py +3 -3
README.md CHANGED
@@ -62,7 +62,7 @@ For the Triton kernels these models were trained with -- much faster, and the
62
  exact code path of the training runs -- install the fork:
63
 
64
  ```bash
65
- pip install git+https://github.com/automl/ComplexKDA
66
  ```
67
 
68
  It is picked up automatically when importable. `COMPLEX_KDA_BACKEND=torch`
 
62
  exact code path of the training runs -- install the fork:
63
 
64
  ```bash
65
+ pip install git+https://github.com/OpenEuroLLM/ComplexKDA
66
  ```
67
 
68
  It is picked up automatically when importable. `COMPLEX_KDA_BACKEND=torch`
modeling_complex_kda.py CHANGED
@@ -13,7 +13,7 @@ else installed.
13
 
14
  IT GOES FASTER WITH THE FORK. When `fla` from
15
 
16
- https://github.com/automl/ComplexKDA
17
 
18
  is importable, the Triton kernels and fused modules it ships are used instead,
19
  and the model is then running exactly the code the checkpoints were trained
@@ -117,7 +117,7 @@ if _REQUESTED == "kernel" and not HAS_KERNEL:
117
  raise ImportError(
118
  "COMPLEX_KDA_BACKEND=kernel, but the ComplexKDA fla fork's signed kernels are not "
119
  "available (need a CUDA device, triton, and `pip install "
120
- "git+https://github.com/automl/ComplexKDA`).")
121
  USE_KERNEL = HAS_KERNEL and _REQUESTED != "torch"
122
 
123
  # Chunk length of the portable recurrence. It trades memory for sequential
@@ -133,7 +133,7 @@ if not USE_KERNEL:
133
  logger.warning_once(
134
  "ComplexKDA is running its portable torch implementation. For the Triton kernels the "
135
  "models were trained with, install the fork: "
136
- "`pip install git+https://github.com/automl/ComplexKDA` (and set "
137
  "COMPLEX_KDA_BACKEND=torch to keep this path).")
138
 
139
 
 
13
 
14
  IT GOES FASTER WITH THE FORK. When `fla` from
15
 
16
+ https://github.com/OpenEuroLLM/ComplexKDA
17
 
18
  is importable, the Triton kernels and fused modules it ships are used instead,
19
  and the model is then running exactly the code the checkpoints were trained
 
117
  raise ImportError(
118
  "COMPLEX_KDA_BACKEND=kernel, but the ComplexKDA fla fork's signed kernels are not "
119
  "available (need a CUDA device, triton, and `pip install "
120
+ "git+https://github.com/OpenEuroLLM/ComplexKDA`).")
121
  USE_KERNEL = HAS_KERNEL and _REQUESTED != "torch"
122
 
123
  # Chunk length of the portable recurrence. It trades memory for sequential
 
133
  logger.warning_once(
134
  "ComplexKDA is running its portable torch implementation. For the Triton kernels the "
135
  "models were trained with, install the fork: "
136
+ "`pip install git+https://github.com/OpenEuroLLM/ComplexKDA` (and set "
137
  "COMPLEX_KDA_BACKEND=torch to keep this path).")
138
 
139