rhaymison commited on
Commit
2013059
·
verified ·
1 Parent(s): 41ca4e6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -4
README.md CHANGED
@@ -62,16 +62,16 @@ verdadeiras. Responda com detalhes e riquesas de informação
62
  <start_of_turn>Me conte sobre o Folclore Nordestino?<end_of_turn>
63
  <start_of_turn>model"""
64
 
65
- # Define the device for model inference
66
  device = "cuda:0"
67
 
68
- # Tokenize the input text
69
  inputs = tokenizer2(text, return_tensors="pt").to(device)
70
 
71
- # Generate output based on the input
72
  outputs = model2.generate(**inputs, max_new_tokens=100, do_sample=False)
73
 
74
- # Decode and print the generated output
75
  output = tokenizer2.decode(outputs[0], skip_special_tokens=True, skip_prompt=True)
76
  print(output.replace("model"," "))
77
 
 
62
  <start_of_turn>Me conte sobre o Folclore Nordestino?<end_of_turn>
63
  <start_of_turn>model"""
64
 
65
+
66
  device = "cuda:0"
67
 
68
+
69
  inputs = tokenizer2(text, return_tensors="pt").to(device)
70
 
71
+
72
  outputs = model2.generate(**inputs, max_new_tokens=100, do_sample=False)
73
 
74
+
75
  output = tokenizer2.decode(outputs[0], skip_special_tokens=True, skip_prompt=True)
76
  print(output.replace("model"," "))
77