Spaces:
Runtime error
Runtime error
Commit
·
fd4433f
1
Parent(s):
e8395e2
update app.py
Browse files
app.py
CHANGED
@@ -17,7 +17,8 @@ def load_model(model_path):
|
|
17 |
model = AutoModelForCausalLM.from_pretrained(
|
18 |
config.base_model_name_or_path,
|
19 |
quantization_config=bnb_config,
|
20 |
-
trust_remote_code=True
|
|
|
21 |
)
|
22 |
model.config.use_cache = False
|
23 |
tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
|
|
|
17 |
model = AutoModelForCausalLM.from_pretrained(
|
18 |
config.base_model_name_or_path,
|
19 |
quantization_config=bnb_config,
|
20 |
+
trust_remote_code=True,
|
21 |
+
token=os.environ["HF_TOKEN"],
|
22 |
)
|
23 |
model.config.use_cache = False
|
24 |
tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
|