Spaces:
Runtime error
Runtime error
fix: minor fix
Browse files
app.py
CHANGED
|
@@ -104,7 +104,7 @@ if torch.cuda.is_available():
|
|
| 104 |
pipe = pipe.to("cuda")
|
| 105 |
|
| 106 |
def get_model_list():
|
| 107 |
-
return models
|
| 108 |
|
| 109 |
unet_cache = {
|
| 110 |
base_name: unet
|
|
|
|
| 104 |
pipe = pipe.to("cuda")
|
| 105 |
|
| 106 |
def get_model_list():
|
| 107 |
+
return models + alt_models
|
| 108 |
|
| 109 |
unet_cache = {
|
| 110 |
base_name: unet
|