curiousily
/

tiny-crypto-sentiment-analysis

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

curiousily commited on Feb 13

Commit

f0b6f68

•

1 Parent(s): b63e376

Update README.md

Files changed (1) hide show

README.md +1 -2

README.md CHANGED Viewed

@@ -27,8 +27,7 @@ tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, use_fast=True)
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_NAME,
     device_map="auto",
-    torch_dtype=torch.float16,
-    trust_remote_code=True,
 )
 pipe = pipeline(

 model = AutoModelForCausalLM.from_pretrained(
     MODEL_NAME,
     device_map="auto",
+    torch_dtype=torch.float16
 )
 pipe = pipeline(