Update README.md
Browse files
README.md
CHANGED
|
@@ -32,7 +32,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_path)
|
|
| 32 |
model = AutoModelForCausalLM.from_pretrained(model_path)
|
| 33 |
inputs = tokenizer(input_text, return_tensors="pt").to(model.device)
|
| 34 |
with torch.no_grad():
|
| 35 |
-
outputs = model.generate(**inputs)
|
| 36 |
```
|
| 37 |
Or you can just use .safetensors weights wherever you want.
|
| 38 |
|
|
|
|
| 32 |
model = AutoModelForCausalLM.from_pretrained(model_path)
|
| 33 |
inputs = tokenizer(input_text, return_tensors="pt").to(model.device)
|
| 34 |
with torch.no_grad():
|
| 35 |
+
outputs = model.generate(**inputs)
|
| 36 |
```
|
| 37 |
Or you can just use .safetensors weights wherever you want.
|
| 38 |
|