Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ model_name = 'TencentARC/Mistral_Pro_8B_v0.1'
|
|
19 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
20 |
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")
|
21 |
# model.generation_config = GenerationConfig.from_pretrained(model_name)
|
22 |
-
model.generation_config.pad_token_id = model.generation_config.eos_token_id
|
23 |
|
24 |
@torch.inference_mode()
|
25 |
@spaces.GPU
|
|
|
19 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
20 |
model = AutoModelForCausalLM.from_pretrained(model_name, torch_dtype=torch.bfloat16, device_map="auto")
|
21 |
# model.generation_config = GenerationConfig.from_pretrained(model_name)
|
22 |
+
# model.generation_config.pad_token_id = model.generation_config.eos_token_id
|
23 |
|
24 |
@torch.inference_mode()
|
25 |
@spaces.GPU
|