diffusers
/

custom-block-template

Model card Files Files and versions

YiYiXu HF Staff commited on Feb 8

Commit

2aca20b

·

verified ·

1 Parent(s): e185079

Update block.py

Files changed (1) hide show

block.py +1 -1

block.py CHANGED Viewed

@@ -30,7 +30,7 @@ class MyCustomBlock(ModularPipelineBlocks):
             # ComponentSpec(
             #     name="model",
             #     type_hint=SomeModelClass,
-            #     repo="organization/model-name",
             # ),
         ]

             # ComponentSpec(
             #     name="model",
             #     type_hint=SomeModelClass,
+            #     pretrained_model_name_or_path="organization/model-name",
             # ),
         ]