Spaces:
Runtime error
Runtime error
TimS commited on
Commit ·
4e6d581
1
Parent(s): 7b2c744
bug fix
Browse files
app.py
CHANGED
|
@@ -42,7 +42,7 @@ args = parser.parse_args()
|
|
| 42 |
|
| 43 |
if HF_SPACE:
|
| 44 |
model = FragLMLlamaModel.from_pretrained(HF_MODEL, token=os.getenv('HF_TOKEN')).to(device)
|
| 45 |
-
designer = FragLMDesign(model=model, tokenizer=
|
| 46 |
else:
|
| 47 |
model = FragLMLlamaModel.from_pretrained(args.model_path).to(device)
|
| 48 |
designer = FragLMDesign(model=model, tokenizer=args.tokenizer_path)
|
|
|
|
| 42 |
|
| 43 |
if HF_SPACE:
|
| 44 |
model = FragLMLlamaModel.from_pretrained(HF_MODEL, token=os.getenv('HF_TOKEN')).to(device)
|
| 45 |
+
designer = FragLMDesign(model=model, tokenizer="/data/fraglm/tokenizer/fraglm_2406_bpe_8k.json")
|
| 46 |
else:
|
| 47 |
model = FragLMLlamaModel.from_pretrained(args.model_path).to(device)
|
| 48 |
designer = FragLMDesign(model=model, tokenizer=args.tokenizer_path)
|