Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -40,7 +40,7 @@ model = loaded_model_name = tokenizer = image_processor = context_len = my_gener
|
|
40 |
def load_model_async(model_path, model_name):
|
41 |
global tokenizer, model, image_processor, context_len, loaded_model_name, my_generator
|
42 |
print(f"Loading {model_name} model ... ")
|
43 |
-
tokenizer, model, image_processor, context_len = load_pretrained_model(model_path, None, model_name)
|
44 |
if "llava" in model_name.lower():
|
45 |
loaded_model_name = "LLaVA"
|
46 |
else:
|
|
|
40 |
def load_model_async(model_path, model_name):
|
41 |
global tokenizer, model, image_processor, context_len, loaded_model_name, my_generator
|
42 |
print(f"Loading {model_name} model ... ")
|
43 |
+
tokenizer, model, image_processor, context_len = load_pretrained_model(model_path, None, model_name, load_4bit=True)
|
44 |
if "llava" in model_name.lower():
|
45 |
loaded_model_name = "LLaVA"
|
46 |
else:
|