jdesiree commited on
Commit
10e8dad
·
verified ·
1 Parent(s): 959f565

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -396,7 +396,8 @@ class Phi3MiniEducationalLLM(Runnable):
396
  repetition_penalty=1.1,
397
  pad_token_id=self.tokenizer.eos_token_id,
398
  early_stopping=True,
399
- use_cache=True # Enable cache for performance
 
400
  )
401
 
402
  # Decode only new tokens
@@ -467,7 +468,8 @@ class Phi3MiniEducationalLLM(Runnable):
467
  "repetition_penalty": 1.2,
468
  "pad_token_id": self.tokenizer.eos_token_id,
469
  "streamer": streamer,
470
- "use_cache": True
 
471
  }
472
 
473
  # Start generation in background
 
396
  repetition_penalty=1.1,
397
  pad_token_id=self.tokenizer.eos_token_id,
398
  early_stopping=True,
399
+ use_cache=False
400
+ past_key_values=None
401
  )
402
 
403
  # Decode only new tokens
 
468
  "repetition_penalty": 1.2,
469
  "pad_token_id": self.tokenizer.eos_token_id,
470
  "streamer": streamer,
471
+ "use_cache": False
472
+ "past_key_values": None
473
  }
474
 
475
  # Start generation in background