Spaces:

bradarrML
/

EleutherAI-gpt-j-6B

Runtime error

Bradarr commited on Dec 12, 2022

Commit

60c7070

1 Parent(s): ed44d2c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -22,7 +22,7 @@ from pprint import pprint
 #     "EleutherAI/gpt-j-6B", revision="float16", torch_dtype=torch.float16, low_cpu_mem_usage=True
 # )
-generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')
 # from transformers import GPTJForCausalLM, AutoTokenizer
 # import torch

 #     "EleutherAI/gpt-j-6B", revision="float16", torch_dtype=torch.float16, low_cpu_mem_usage=True
 # )
+generator = pipeline('text-generation', model='EleutherAI/gpt-neo-1.3B')
 # from transformers import GPTJForCausalLM, AutoTokenizer
 # import torch