Update README.md

README.md

@@ -43,7 +43,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, 1girl, ahoge, braid, closed eyes, collared dress, dress, flower, full body, hair flower, hair ornament, long hair, night, night sky, outdoors, parted lips, pink flower, pink hair, short sleeves, sky, solo, straight hair, sunflower, very long hair, white flower
@@ -113,7 +113,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))
 ```