diffusers/sdxl-to-diffusers
28
1o
2 H��d}3 � @ s� d dl Zd dlZd dlZd dlZd dlmZ d dlmZ d dl m4Z5 d dlm Z d dl Z d dl
mZ d dlmZmZmZmZ d dlmZ d d lmZ d d6lmZ d Zd ed
ededefdd�Z ddd edede d fdd�Z e� � fded ed
efdd�Z dS ) � N)�Path)�Any)�TemporaryDirectory)�Optional)�BytesIO)�7CommitInfo�8Discussion�HfApi�hf_hub_download)�repo_folder_name)�StableDiffusionXLPipeline)�CONFIG_MAPPINGz? This PR adds fp32 and fp16 weights in safetensors format to {}�model_id�filename�folder�progressc C sp |ddd� t j�| |�}t j�|�r|nt| |d�}t�|�}|j|dd� |jt j9d�}|j|ddd � |S )10Nr zDownloading model��desc)�repo_idr T)�safe_serialization)� torch_dtype�fp16)r �variant) �os�path�join�isfiler11 r �from_single_file�save_pretrained�to�torch�float16)r r r r �12local_file� ckpt_file�pipeline� r% �*/home/patrick/sdxl-to-diffusers/convert.py�convert_single s 13r' �apir �pr_title�returnr c C sp z| j |d�}W n14 ty Y d S w |D ] }|jdkr5|jr5|j|kr5| j||jd�}|jdkr5| S qd S )N)r �open)r �discussion_numzrefs/heads/main)�get_repo_discussions� Exception�status�is_pull_request�title�get_discussion_details�num�
target_branch)r( r r) � discussions�15discussion�detailsr% r% r& � previous_pr$ s �16��r8 �tokenc C s t � }d}t� �}}tj�|t|dd��}t�|� d }zZz>t||||�}|ddd� |j|d|d| |t �17|�d d18�}|� d �d � d
�d } ddtj�|d| � � �}19|ddd� W n t yn } zt
j�t| ���d } ~ ww W t�|� nt�|� w |20W d � S 1 s�w Y d S )Nz(Adding `diffusers` weights of this model�models)r � repo_typegffffff�?zUploading to Hubr z./�modelT)� folder_path� path_in_repor r; r9 �commit_message�commit_description� create_prz%2F������/r zPr created at: zhttps://huggingface.co/r5 � �Done)r r r r r r �makedirsr' �
upload_folder�COMMIT_MESSAGE�format�splitr. �gr�21exceptions�Error�str�shutil�rmtree) r9 r r r r( r) �dr �new_pr� pr_number�link�er% r% r& �convert0 s* 22 ���$�rV ) �gradiorK �requestsr rO �pathlibr �typingr �tempfiler r r �ior �huggingface_hubr r r r23 � huggingface_hub.file_downloadr � diffusersr � transformersr
rH rN r' r8 �ProgressrV r% r% r% r& �<module> s$ "