p1atdev commited on
Commit
dd5a3f3
1 Parent(s): c06e5da

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -43,7 +43,7 @@ prompt = "<|bos|><rating>rating:sfw, rating:general</rating><copyright>original<
43
  inputs = tokenizer(prompt, return_tensors="pt").input_ids
44
 
45
  with torch.no_grad():
46
- outputs = model.generate(inputs, generation_config=generation_config)
47
 
48
  print(tokenizer.decode(outputs[0], skip_special_tokens=True))
49
  # 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({
113
  }, return_tensors="pt", tokenize=True)
114
 
115
  with torch.no_grad():
116
- outputs = ort_model.generate(inputs, generation_config=generation_config)
117
 
118
  print(tokenizer.decode(outputs[0], skip_special_tokens=True))
119
  ```
 
43
  inputs = tokenizer(prompt, return_tensors="pt").input_ids
44
 
45
  with torch.no_grad():
46
+ outputs = model.generate(inputs, generation_config=model.generation_config)
47
 
48
  print(tokenizer.decode(outputs[0], skip_special_tokens=True))
49
  # 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
  }, return_tensors="pt", tokenize=True)
114
 
115
  with torch.no_grad():
116
+ outputs = ort_model.generate(inputs, generation_config=model.generation_config)
117
 
118
  print(tokenizer.decode(outputs[0], skip_special_tokens=True))
119
  ```