Update README.md

README.md

@@ -42,7 +42,7 @@ prompt = "<|bos|><rating>rating:sfw, rating:general</rating><copyright>original<
 inputs = tokenizer(prompt, return_tensors="pt").input_ids
 
 with torch.no_grad():
-  outputs = model.generate(inputs, generation_config=generation_config)
+  outputs = model.generate(inputs, generation_config=model.generation_config)
 
 print(tokenizer.decode(outputs[0], skip_special_tokens=True))
 # rating:sfw, rating:general, original, 1girl, ahoge, black hair, blue eyes, blush, closed mouth, ear piercing, earrings, jewelry, looking at viewer, mole, mole under eye, piercing, portrait, shirt, short hair, solo, white shirt
@@ -111,7 +111,7 @@ inputs = tokenizer.apply_chat_template({
 }, return_tensors="pt", tokenize=True,)
 
 with torch.no_grad():
-  outputs = ort_model.generate(inputs, generation_config=generation_config)
+  outputs = ort_model.generate(inputs, generation_config=model.generation_config)
 
 print(tokenizer.decode(outputs[0], skip_special_tokens=True))
 ```