Upload 18 files
Browse files
utils/convert_url_to_diffusers_sd/convert_url_to_diffusers_sd.py
CHANGED
@@ -234,7 +234,7 @@ def convert_url_to_diffusers_sd(url, civitai_key="", half=True, vae=None, schedu
|
|
234 |
config_url = "https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference-v.yaml"
|
235 |
|
236 |
pipe = None
|
237 |
-
if is_repo_name(
|
238 |
if half:
|
239 |
pipe = StableDiffusionPipeline.from_pretrained(new_file, extract_ema=extract_ema, requires_safety_checker=False, use_safetensors=True, torch_dtype=torch.float16)
|
240 |
else:
|
|
|
234 |
config_url = "https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/v2-inference-v.yaml"
|
235 |
|
236 |
pipe = None
|
237 |
+
if is_repo_name(url):
|
238 |
if half:
|
239 |
pipe = StableDiffusionPipeline.from_pretrained(new_file, extract_ema=extract_ema, requires_safety_checker=False, use_safetensors=True, torch_dtype=torch.float16)
|
240 |
else:
|
utils/convert_url_to_diffusers_sdxl/convert_url_to_diffusers_sdxl.py
CHANGED
@@ -223,7 +223,7 @@ def convert_url_to_diffusers_sdxl(url, civitai_key="", half=True, vae=None, sche
|
|
223 |
new_repo_name = Path(new_file).stem.replace(" ", "_").replace(",", "_").replace(".", "_") #
|
224 |
|
225 |
pipe = None
|
226 |
-
if is_repo_name(
|
227 |
if half:
|
228 |
pipe = StableDiffusionXLPipeline.from_pretrained(new_file, use_safetensors=True, torch_dtype=torch.float16)
|
229 |
else:
|
|
|
223 |
new_repo_name = Path(new_file).stem.replace(" ", "_").replace(",", "_").replace(".", "_") #
|
224 |
|
225 |
pipe = None
|
226 |
+
if is_repo_name(url):
|
227 |
if half:
|
228 |
pipe = StableDiffusionXLPipeline.from_pretrained(new_file, use_safetensors=True, torch_dtype=torch.float16)
|
229 |
else:
|