Instructions to use hf-internal-testing/tiny-helios-pyramid-modular-pipe with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Diffusers
How to use hf-internal-testing/tiny-helios-pyramid-modular-pipe with Diffusers:
pip install -U diffusers transformers accelerate
import torch from diffusers import DiffusionPipeline # switch to "mps" for apple devices pipe = DiffusionPipeline.from_pretrained("hf-internal-testing/tiny-helios-pyramid-modular-pipe", dtype=torch.bfloat16, device_map="cuda") prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" image = pipe(prompt).images[0] - Notebooks
- Google Colab
- Kaggle
Update modular_model_index.json
Browse files- modular_model_index.json +5 -5
modular_model_index.json
CHANGED
|
@@ -6,7 +6,7 @@
|
|
| 6 |
"diffusers",
|
| 7 |
"HeliosScheduler",
|
| 8 |
{
|
| 9 |
-
"pretrained_model_name_or_path": "
|
| 10 |
"revision": null,
|
| 11 |
"subfolder": "scheduler",
|
| 12 |
"type_hint": [
|
|
@@ -20,7 +20,7 @@
|
|
| 20 |
"transformers",
|
| 21 |
"UMT5EncoderModel",
|
| 22 |
{
|
| 23 |
-
"pretrained_model_name_or_path": "
|
| 24 |
"revision": null,
|
| 25 |
"subfolder": "text_encoder",
|
| 26 |
"type_hint": [
|
|
@@ -34,7 +34,7 @@
|
|
| 34 |
"transformers",
|
| 35 |
"T5TokenizerFast",
|
| 36 |
{
|
| 37 |
-
"pretrained_model_name_or_path": "
|
| 38 |
"revision": null,
|
| 39 |
"subfolder": "tokenizer",
|
| 40 |
"type_hint": [
|
|
@@ -48,7 +48,7 @@
|
|
| 48 |
"diffusers",
|
| 49 |
"HeliosTransformer3DModel",
|
| 50 |
{
|
| 51 |
-
"pretrained_model_name_or_path": "
|
| 52 |
"revision": null,
|
| 53 |
"subfolder": "transformer",
|
| 54 |
"type_hint": [
|
|
@@ -62,7 +62,7 @@
|
|
| 62 |
"diffusers",
|
| 63 |
"AutoencoderKLWan",
|
| 64 |
{
|
| 65 |
-
"pretrained_model_name_or_path": "
|
| 66 |
"revision": null,
|
| 67 |
"subfolder": "vae",
|
| 68 |
"type_hint": [
|
|
|
|
| 6 |
"diffusers",
|
| 7 |
"HeliosScheduler",
|
| 8 |
{
|
| 9 |
+
"pretrained_model_name_or_path": "hf-internal-testing/tiny-helios-pyramid-modular-pipe",
|
| 10 |
"revision": null,
|
| 11 |
"subfolder": "scheduler",
|
| 12 |
"type_hint": [
|
|
|
|
| 20 |
"transformers",
|
| 21 |
"UMT5EncoderModel",
|
| 22 |
{
|
| 23 |
+
"pretrained_model_name_or_path": "hf-internal-testing/tiny-helios-pyramid-modular-pipe",
|
| 24 |
"revision": null,
|
| 25 |
"subfolder": "text_encoder",
|
| 26 |
"type_hint": [
|
|
|
|
| 34 |
"transformers",
|
| 35 |
"T5TokenizerFast",
|
| 36 |
{
|
| 37 |
+
"pretrained_model_name_or_path": "hf-internal-testing/tiny-helios-pyramid-modular-pipe",
|
| 38 |
"revision": null,
|
| 39 |
"subfolder": "tokenizer",
|
| 40 |
"type_hint": [
|
|
|
|
| 48 |
"diffusers",
|
| 49 |
"HeliosTransformer3DModel",
|
| 50 |
{
|
| 51 |
+
"pretrained_model_name_or_path": "hf-internal-testing/tiny-helios-pyramid-modular-pipe",
|
| 52 |
"revision": null,
|
| 53 |
"subfolder": "transformer",
|
| 54 |
"type_hint": [
|
|
|
|
| 62 |
"diffusers",
|
| 63 |
"AutoencoderKLWan",
|
| 64 |
{
|
| 65 |
+
"pretrained_model_name_or_path": "hf-internal-testing/tiny-helios-pyramid-modular-pipe",
|
| 66 |
"revision": null,
|
| 67 |
"subfolder": "vae",
|
| 68 |
"type_hint": [
|