Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -56,7 +56,7 @@ llva_tokenizer, llva_model, llva_image_processor, llva_context_len = load_pretra
|
|
56 |
model_name=get_model_name_from_path(llva_model_path),)
|
57 |
# llva_tokenizer.cuda()
|
58 |
llva_model.to(device)
|
59 |
-
llva_image_processor.to(device)
|
60 |
|
61 |
### Load consistentID_model checkpoint
|
62 |
pipe.load_ConsistentID_model(
|
|
|
56 |
model_name=get_model_name_from_path(llva_model_path),)
|
57 |
# llva_tokenizer.cuda()
|
58 |
llva_model.to(device)
|
59 |
+
# llva_image_processor.to(device)
|
60 |
|
61 |
### Load consistentID_model checkpoint
|
62 |
pipe.load_ConsistentID_model(
|