Spaces:
Running
Running
Update convert.py
Browse files- convert.py +9 -4
convert.py
CHANGED
|
@@ -38,15 +38,20 @@ def convert_single(model_id: str, token:str, filename: str, model_type: str, sam
|
|
| 38 |
config_url = (Path(model_id)/"resolve/main"/filename).with_suffix(".yaml")
|
| 39 |
config_url = "https://huggingface.co/" + str(config_url)
|
| 40 |
|
| 41 |
-
config_file = BytesIO(requests.get(config_url).content)
|
| 42 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 43 |
if model_type == "ControlNet":
|
| 44 |
progress(0.2, desc="Converting ControlNet Model")
|
| 45 |
-
pipeline = download_controlnet_from_original_ckpt(ckpt_file,
|
| 46 |
to_args = {"dtype": torch.float16}
|
| 47 |
else:
|
| 48 |
progress(0.1, desc="Converting Model")
|
| 49 |
-
pipeline = download_from_original_stable_diffusion_ckpt(ckpt_file,
|
| 50 |
to_args = {"torch_dtype": torch.float16}
|
| 51 |
|
| 52 |
pipeline.save_pretrained(folder)
|
|
|
|
| 38 |
config_url = (Path(model_id)/"resolve/main"/filename).with_suffix(".yaml")
|
| 39 |
config_url = "https://huggingface.co/" + str(config_url)
|
| 40 |
|
| 41 |
+
#config_file = BytesIO(requests.get(config_url).content)
|
| 42 |
+
|
| 43 |
+
response = requests.get(config_url)
|
| 44 |
+
with tempfile.NamedTemporaryFile(delete=False, mode='wb') as tmp_file:
|
| 45 |
+
tmp_file.write(response.content)
|
| 46 |
+
temp_config_file_path = tmp_file.name
|
| 47 |
+
|
| 48 |
if model_type == "ControlNet":
|
| 49 |
progress(0.2, desc="Converting ControlNet Model")
|
| 50 |
+
pipeline = download_controlnet_from_original_ckpt(ckpt_file, temp_config_file_path, image_size=sample_size, from_safetensors=from_safetensors, extract_ema=extract_ema)
|
| 51 |
to_args = {"dtype": torch.float16}
|
| 52 |
else:
|
| 53 |
progress(0.1, desc="Converting Model")
|
| 54 |
+
pipeline = download_from_original_stable_diffusion_ckpt(ckpt_file, temp_config_file_path, image_size=sample_size, scheduler_type=scheduler_type, from_safetensors=from_safetensors, extract_ema=extract_ema)
|
| 55 |
to_args = {"torch_dtype": torch.float16}
|
| 56 |
|
| 57 |
pipeline.save_pretrained(folder)
|