intervention-demo-finetuned

Sleeping

yilunzhao commited on Nov 7

Commit

026f5b2

•

1 Parent(s): 7eb73dd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,7 +6,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer
 # Load model and tokenizer if a GPU is available
 if torch.cuda.is_available():
-    model_id = "allenai/OLMo-7B-Instruct"
     model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", trust_remote_code=True)
     tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
 else:

 # Load model and tokenizer if a GPU is available
 if torch.cuda.is_available():
+    model_id = "yilunzhao/olmo-finetuned"
     model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto", trust_remote_code=True)
     tokenizer = AutoTokenizer.from_pretrained(model_id, trust_remote_code=True)
 else: