Update model.py

model.py

@@ -6,19 +6,15 @@ from transformers import AutoConfig, AutoModelForCausalLM, AutoTokenizer, TextIt
 
 model_id = 'abacaj/starcoderbase-1b-sft'
 
-config = AutoConfig.from_pretrained(model_id)
-config.pretraining_tp = 1
 if torch.cuda.is_available():
     model = AutoModelForCausalLM.from_pretrained(
         model_id,
-        config=config,
         torch_dtype=torch.float16,
         device_map='cuda',
     )
 else:
     model = AutoModelForCausalLM.from_pretrained(
         model_id,
-        config=config,
         torch_dtype=torch.float32,
     )
 tokenizer = AutoTokenizer.from_pretrained(model_id)