Update app.py
Browse files
app.py
CHANGED
|
@@ -25,6 +25,9 @@ def download_lora(url):
|
|
| 25 |
|
| 26 |
# If file already exists in cache, return the path
|
| 27 |
if os.path.exists(local_path):
|
|
|
|
|
|
|
|
|
|
| 28 |
return local_path
|
| 29 |
|
| 30 |
# Download the file
|
|
@@ -43,10 +46,14 @@ def download_lora(url):
|
|
| 43 |
for chunk in response.iter_content(chunk_size=8192):
|
| 44 |
if chunk:
|
| 45 |
f.write(chunk)
|
| 46 |
-
|
|
|
|
|
|
|
| 47 |
return local_path
|
| 48 |
except Exception as e:
|
|
|
|
| 49 |
print(f"Error downloading LoRA: {str(e)}")
|
|
|
|
| 50 |
return None
|
| 51 |
|
| 52 |
def is_civitai_url(url):
|
|
@@ -100,17 +107,33 @@ def generate_image(prompt, negative_prompt, lora_url, num_inference_steps=30, gu
|
|
| 100 |
lora_path = download_lora(lora_url)
|
| 101 |
if lora_path:
|
| 102 |
pipe.load_lora_weights(lora_path)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 103 |
else:
|
| 104 |
pipe.load_lora_weights(DEFAULT_LORA)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 105 |
# If it's a HuggingFace repo path
|
| 106 |
elif '/' in lora_url and not lora_url.startswith('http'):
|
| 107 |
pipe.load_lora_weights(lora_url)
|
|
|
|
|
|
|
|
|
|
| 108 |
else:
|
| 109 |
pipe.load_lora_weights(DEFAULT_LORA)
|
|
|
|
|
|
|
|
|
|
| 110 |
else:
|
| 111 |
pipe.load_lora_weights(DEFAULT_LORA)
|
| 112 |
except Exception as e:
|
|
|
|
| 113 |
print(f"Error loading LoRA weights: {str(e)}")
|
|
|
|
| 114 |
pipe.load_lora_weights(DEFAULT_LORA)
|
| 115 |
|
| 116 |
if model == "Real6.0":
|
|
|
|
| 25 |
|
| 26 |
# If file already exists in cache, return the path
|
| 27 |
if os.path.exists(local_path):
|
| 28 |
+
print()
|
| 29 |
+
print("********** Lora Already Exists **********")
|
| 30 |
+
print()
|
| 31 |
return local_path
|
| 32 |
|
| 33 |
# Download the file
|
|
|
|
| 46 |
for chunk in response.iter_content(chunk_size=8192):
|
| 47 |
if chunk:
|
| 48 |
f.write(chunk)
|
| 49 |
+
print()
|
| 50 |
+
print("********** Lora Downloading Successfull **********")
|
| 51 |
+
print()
|
| 52 |
return local_path
|
| 53 |
except Exception as e:
|
| 54 |
+
print()
|
| 55 |
print(f"Error downloading LoRA: {str(e)}")
|
| 56 |
+
print()
|
| 57 |
return None
|
| 58 |
|
| 59 |
def is_civitai_url(url):
|
|
|
|
| 107 |
lora_path = download_lora(lora_url)
|
| 108 |
if lora_path:
|
| 109 |
pipe.load_lora_weights(lora_path)
|
| 110 |
+
print()
|
| 111 |
+
print("********** URL Lora Loaded **********")
|
| 112 |
+
print()
|
| 113 |
+
|
| 114 |
else:
|
| 115 |
pipe.load_lora_weights(DEFAULT_LORA)
|
| 116 |
+
print()
|
| 117 |
+
print("********** Default Lora Loaded **********")
|
| 118 |
+
print()
|
| 119 |
+
|
| 120 |
# If it's a HuggingFace repo path
|
| 121 |
elif '/' in lora_url and not lora_url.startswith('http'):
|
| 122 |
pipe.load_lora_weights(lora_url)
|
| 123 |
+
print()
|
| 124 |
+
print("********** URL Lora Loaded **********")
|
| 125 |
+
print()
|
| 126 |
else:
|
| 127 |
pipe.load_lora_weights(DEFAULT_LORA)
|
| 128 |
+
print()
|
| 129 |
+
print("********** Default Lora Loaded **********")
|
| 130 |
+
print()
|
| 131 |
else:
|
| 132 |
pipe.load_lora_weights(DEFAULT_LORA)
|
| 133 |
except Exception as e:
|
| 134 |
+
print()
|
| 135 |
print(f"Error loading LoRA weights: {str(e)}")
|
| 136 |
+
print()
|
| 137 |
pipe.load_lora_weights(DEFAULT_LORA)
|
| 138 |
|
| 139 |
if model == "Real6.0":
|