Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -9,7 +9,9 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
| 9 |
# List of models
|
| 10 |
models = {
|
| 11 |
"sdxl-turbo": "stabilityai/sdxl-turbo",
|
| 12 |
-
"MistoLine": "TheMistoAI/MistoLine"
|
|
|
|
|
|
|
| 13 |
}
|
| 14 |
|
| 15 |
# Cache to store loaded pipelines
|
|
|
|
| 9 |
# List of models
|
| 10 |
models = {
|
| 11 |
"sdxl-turbo": "stabilityai/sdxl-turbo",
|
| 12 |
+
"MistoLine": "TheMistoAI/MistoLine",
|
| 13 |
+
"UnfilteredAI/NSFW": "UnfilteredAI/NSFW-gen-v2",
|
| 14 |
+
"runwayml/SD":"runwayml/stable-diffusion-v1-5"
|
| 15 |
}
|
| 16 |
|
| 17 |
# Cache to store loaded pipelines
|