OpenAssistant
/

galactica-6.7b-finetuned

Text Generation

Inference Endpoints

text-generation-inference

Model card Files Files and versions Community

sanagnos commited on Jan 16, 2023

Commit

8cd997d

•

1 Parent(s): 618f20f

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -8,7 +8,7 @@ from torch import nn
 from torch.nn import functional as F
 import transformers
-base_path = 'sanagnos/galactica-6.7b-finetuned'
 model = transformers.OPTForCausalLM.from_pretrained(
     base_path, load_in_8bit=True, device_map='auto', low_cpu_mem_usage=True,

 from torch.nn import functional as F
 import transformers
+base_path = 'OpenAssistant/galactica-6.7b-finetuned'
 model = transformers.OPTForCausalLM.from_pretrained(
     base_path, load_in_8bit=True, device_map='auto', low_cpu_mem_usage=True,