Update vision.py
This commit is contained in:
parent
60b3da5fe5
commit
19c69280b7
1 changed files with 1 additions and 0 deletions
|
|
@ -289,6 +289,7 @@ class FastBaseModel:
|
|||
# Cannot be None, since HF now checks for the config
|
||||
if load_in_4bit: kwargs["quantization_config"] = bnb_config
|
||||
|
||||
print(model_name)
|
||||
model = auto_model.from_pretrained(
|
||||
model_name,
|
||||
device_map = device_map,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue