Spaces:
Running
on
Zero
Running
on
Zero
rafaaa2105
commited on
Commit
•
c576f11
1
Parent(s):
c92c1fc
Update app.py
Browse files
app.py
CHANGED
@@ -13,7 +13,7 @@ print(f"Detected {len(model_list)} on models and {len(lora_list)} LoRAs.")
|
|
13 |
models = {}
|
14 |
for model_name in model_list:
|
15 |
try:
|
16 |
-
print(f"
|
17 |
models[model_name] = DiffusionPipeline.from_pretrained(model_name, torch_dtype=torch.float16).to("cuda")
|
18 |
except Exception as e:
|
19 |
print(f"Error loading model {model_name}: {e}")
|
|
|
13 |
models = {}
|
14 |
for model_name in model_list:
|
15 |
try:
|
16 |
+
print(f"\n\nLoading {model_name}...")
|
17 |
models[model_name] = DiffusionPipeline.from_pretrained(model_name, torch_dtype=torch.float16).to("cuda")
|
18 |
except Exception as e:
|
19 |
print(f"Error loading model {model_name}: {e}")
|