Update pipeline.py

pipeline.py

@@ -7,12 +7,12 @@ class PreTrainedPipeline():
         path = "oleksandrfluxon/mpt-7b-instruct-2"
         print("===> path", path)
         
-        config = transformers.AutoConfig.from_pretrained(name, trust_remote_code=True)
+        config = transformers.AutoConfig.from_pretrained(path, trust_remote_code=True)
         config.max_seq_len = 4096 # (input + output) tokens can now be up to 4096
 
         print("===> loading model")
         model = transformers.AutoModelForCausalLM.from_pretrained(
-          name,
+          path,
           config=config,
           torch_dtype=torch.bfloat16, # Load model weights in bfloat16
           trust_remote_code=True,