pszemraj commited on
Commit
4abf7ef
·
verified ·
1 Parent(s): a6ad7c9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +3 -1
README.md CHANGED
@@ -32,7 +32,9 @@ It achieves the following results on the evaluation set:
32
  from transformers import pipeline
33
 
34
  pipe = pipeline(
35
- "text2text-generation", model="pszemraj/tFINE-850m-24x24-v0.5-instruct-L1"
 
 
36
  )
37
  prompt = "write a python script to download a file from a url and save as a local file using requests. explain how it works"
38
  res = pipe(
 
32
  from transformers import pipeline
33
 
34
  pipe = pipeline(
35
+ "text2text-generation",
36
+ model="pszemraj/tFINE-850m-24x24-v0.5-instruct-L1",
37
+ device_map="auto",
38
  )
39
  prompt = "write a python script to download a file from a url and save as a local file using requests. explain how it works"
40
  res = pipe(