Update model_class.py

model_class.py

@@ -191,7 +191,7 @@ class TicketGPT(
     
         # Prepare inputs to the model
         input_ids = tokenizer.encode(text)
-        supported_context_length = model.pos_emb.weight.shape[0]
+        supported_context_length = self.config.context_length
     
         # Truncate sequences if they too long
         input_ids = input_ids[:min(max_length, supported_context_length)]