Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -225,22 +225,7 @@ TITLE = """
|
|
| 225 |
|
| 226 |
# UPDATED: Description to reflect new workflow
|
| 227 |
DESCRIPTION = """
|
| 228 |
-
This Space allows you to automatically export 🤗 transformers
|
| 229 |
-
|
| 230 |
-
Simply provide a model ID from the Hugging Face Hub, and choose your desired output.
|
| 231 |
-
|
| 232 |
-
### ✨ Key Features
|
| 233 |
-
|
| 234 |
-
* **🚀 Create a New Optimized Repo**: Automatically converts your model and uploads it to a new repository under your username (e.g., `your-username/model-name-neuron`).
|
| 235 |
-
* **🔗 Link Back to Original**: Creates a Pull Request on the original model's repository to add a link to your optimized version, making it easier for the community to discover.
|
| 236 |
-
* **🛠️ PR to a Custom Repo**: For custom workflows, you can create a Pull Request to add the optimized files directly into an existing repository you own.
|
| 237 |
-
* **📦 Contribute to Cache**: Contribute the generated compilation artifacts to a centralized cache repository (or your own private cache), helping avoid recompilation of already exported models.
|
| 238 |
-
|
| 239 |
-
### ⚙️ How to Use
|
| 240 |
-
1. **Model ID**: Enter the ID of the model you want to export (e.g., `bert-base-uncased` or `stabilityai/stable-diffusion-xl-base-1.0`) and choose the corresponding task.
|
| 241 |
-
2. **Export Options**: Select at least one option for where to save the exported model. You can provide your own cache repo ID or use the default (`aws-neuron/optimum-neuron-cache`).
|
| 242 |
-
3. **Convert & Upload**: Click the button and follow the logs to track progress!
|
| 243 |
-
|
| 244 |
"""
|
| 245 |
|
| 246 |
CUSTOM_CSS = """
|
|
@@ -393,6 +378,30 @@ with gr.Blocks(css=CUSTOM_CSS, theme=gr.themes.Soft()) as demo:
|
|
| 393 |
],
|
| 394 |
outputs=log_box,
|
| 395 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 396 |
|
| 397 |
with gr.Tab("Supported Architectures"):
|
| 398 |
gr.HTML(f"""
|
|
|
|
| 225 |
|
| 226 |
# UPDATED: Description to reflect new workflow
|
| 227 |
DESCRIPTION = """
|
| 228 |
+
This Space allows you to automatically export 🤗 transformers to AWS Neuron-optimized format for Inferentia/Trainium acceleration.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 229 |
"""
|
| 230 |
|
| 231 |
CUSTOM_CSS = """
|
|
|
|
| 378 |
],
|
| 379 |
outputs=log_box,
|
| 380 |
)
|
| 381 |
+
|
| 382 |
+
with gr.Tab("Get Started"):
|
| 383 |
+
gr.Markdown(
|
| 384 |
+
"""
|
| 385 |
+
**Optimum-Neuron version used:** 0.4.1
|
| 386 |
+
|
| 387 |
+
This Space allows you to automatically export 🤗 transformers to AWS Neuron-optimized format for Inferentia/Trainium acceleration.
|
| 388 |
+
|
| 389 |
+
Simply provide a model ID from the Hugging Face Hub, and choose your desired output.
|
| 390 |
+
|
| 391 |
+
### ✨ Key Features
|
| 392 |
+
|
| 393 |
+
* **🚀 Create a New Optimized Repo**: Automatically converts your model and uploads it to a new repository under your username (e.g., `your-username/model-name-neuron`).
|
| 394 |
+
* **🔗 Link Back to Original**: Creates a Pull Request on the original model's repository to add a link to your optimized version, making it easier for the community to discover.
|
| 395 |
+
* **🛠️ PR to a Custom Repo**: For custom workflows, you can create a Pull Request to add the optimized files directly into an existing repository you own.
|
| 396 |
+
* **📦 Contribute to Cache**: Contribute the generated compilation artifacts to a centralized cache repository (or your own private cache), helping avoid recompilation of already exported models.
|
| 397 |
+
|
| 398 |
+
### ⚙️ How to Use
|
| 399 |
+
1. **Model ID**: Enter the ID of the model you want to export (e.g., `bert-base-uncased` or `stabilityai/stable-diffusion-xl-base-1.0`) and choose the corresponding task.
|
| 400 |
+
2. **Export Options**: Select at least one option for where to save the exported model. You can provide your own cache repo ID or use the default (`aws-neuron/optimum-neuron-cache`).
|
| 401 |
+
3. **Convert & Upload**: Click the button and follow the logs to track progress!
|
| 402 |
+
"""
|
| 403 |
+
)
|
| 404 |
+
|
| 405 |
|
| 406 |
with gr.Tab("Supported Architectures"):
|
| 407 |
gr.HTML(f"""
|