Spaces:
Paused
Paused
Update app.py
Browse files
app.py
CHANGED
|
@@ -21,10 +21,10 @@ ckpt = hf_hub_download(repo_id="lambdalabs/image-mixer", filename="image-mixer-p
|
|
| 21 |
config = hf_hub_download(repo_id="lambdalabs/image-mixer", filename="image-mixer-config.yaml", cache_dir="/data/.cache")
|
| 22 |
|
| 23 |
device = "cuda:0"
|
| 24 |
-
|
| 25 |
-
|
| 26 |
|
| 27 |
-
|
| 28 |
|
| 29 |
n_inputs = 5
|
| 30 |
|
|
|
|
| 21 |
config = hf_hub_download(repo_id="lambdalabs/image-mixer", filename="image-mixer-config.yaml", cache_dir="/data/.cache")
|
| 22 |
|
| 23 |
device = "cuda:0"
|
| 24 |
+
model = load_model_from_config(config, ckpt, device=device, verbose=False)
|
| 25 |
+
model = model.to(device).half()
|
| 26 |
|
| 27 |
+
clip_model, preprocess = clip.load("ViT-L/14", device=device)
|
| 28 |
|
| 29 |
n_inputs = 5
|
| 30 |
|