1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00
This commit is contained in:
Dhruv Nair
2023-12-22 14:07:54 +00:00
parent 2686fddbf1
commit ef656d73cd

View File

@@ -11,6 +11,7 @@
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import inspect
from contextlib import nullcontext
from io import BytesIO
from pathlib import Path
@@ -54,6 +55,10 @@ MODEL_TYPE_FROM_PIPELINE_CLASS = {
}
def extract_pipeline_compoments(pipeline_class):
components = inspect.signature(pipeline_class).parameters.keys()
return components
def check_valid_url(pretrained_model_link_or_path):
# remove huggingface url
@@ -284,6 +289,7 @@ class FromSingleFileMixin:
f"The provided path is either not a file or a valid huggingface URL was not provided. Valid URLs begin with {', '.join(VALID_URL_PREFIXES)}"
)
pretrained_model_link_or_path = fetch_model_checkpoint(ckpt_path, cache_dir=cache_dir, resume_download=resume_download, proxies=proxies, local_files_only=local_files_only, token=token, revision=revision)
components = extract_pipeline_compoments(cls)
pipe = download_from_original_stable_diffusion_ckpt(
pretrained_model_link_or_path,