badaoui HF Staff commited on
Commit
a03bf63
·
verified ·
1 Parent(s): df6d536

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +25 -16
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 and 🧨 diffusion models to AWS Neuron-optimized format for Inferentia/Trainium acceleration.
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"""