Spaces:
Running
on
A10G
Running
on
A10G
Update app.py
Browse files
app.py
CHANGED
@@ -62,7 +62,7 @@ def _load_model_tokenizer(args):
|
|
62 |
device_map=device_map,
|
63 |
trust_remote_code=True,
|
64 |
resume_download=True,
|
65 |
-
token=YOUR_ACCESS_TOKEN
|
66 |
).eval()
|
67 |
model.generation_config = GenerationConfig.from_pretrained(
|
68 |
args.checkpoint_path, trust_remote_code=True, resume_download=True #, token=YOUR_ACCESS_TOKEN
|
|
|
62 |
device_map=device_map,
|
63 |
trust_remote_code=True,
|
64 |
resume_download=True,
|
65 |
+
# token=YOUR_ACCESS_TOKEN
|
66 |
).eval()
|
67 |
model.generation_config = GenerationConfig.from_pretrained(
|
68 |
args.checkpoint_path, trust_remote_code=True, resume_download=True #, token=YOUR_ACCESS_TOKEN
|