Spaces:
Sleeping
Sleeping
Ben Burtenshaw
commited on
Commit
·
e838258
1
Parent(s):
8e226e6
lose flash attn
Browse files- prometheus_pipeline.py +1 -1
- requirements.txt +1 -2
prometheus_pipeline.py
CHANGED
|
@@ -55,4 +55,4 @@ if __name__ == "__main__":
|
|
| 55 |
print("--- %s seconds ---" % (time.time() - start_time))
|
| 56 |
|
| 57 |
if distiset is not None:
|
| 58 |
-
distiset.push_to_hub("instruction-dataset-prometheus")
|
|
|
|
| 55 |
print("--- %s seconds ---" % (time.time() - start_time))
|
| 56 |
|
| 57 |
if distiset is not None:
|
| 58 |
+
distiset.push_to_hub("instruction-dataset-prometheus")
|
requirements.txt
CHANGED
|
@@ -1,3 +1,2 @@
|
|
| 1 |
gradio[oauth]
|
| 2 |
-
distilabel[vllm] @ git+https://github.com/argilla-io/distilabel.git@develop
|
| 3 |
-
flash-attn
|
|
|
|
| 1 |
gradio[oauth]
|
| 2 |
+
distilabel[vllm] @ git+https://github.com/argilla-io/distilabel.git@develop
|
|
|