Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -3,6 +3,9 @@ import os
|
|
3 |
from repo_utils import extract_repo_content
|
4 |
from model_utils import get_model_summary, install_flash_attn
|
5 |
|
|
|
|
|
|
|
6 |
def format_output(extracted_content, repo_url):
|
7 |
formatted_output = f"# Repository URL: {repo_url}\n\n"
|
8 |
for file_data in extracted_content:
|
|
|
3 |
from repo_utils import extract_repo_content
|
4 |
from model_utils import get_model_summary, install_flash_attn
|
5 |
|
6 |
+
# Install required package
|
7 |
+
install_flash_attn()
|
8 |
+
|
9 |
def format_output(extracted_content, repo_url):
|
10 |
formatted_output = f"# Repository URL: {repo_url}\n\n"
|
11 |
for file_data in extracted_content:
|