Hugging Face's logo

Spaces:
pinyuchen
/
Gradient-Cuff-Jailbreak-Detector-Gemma-2B
Sleeping

App Files Community
2
pinyuchen commited on
Commit
3746678
·
verified ·
1 Parent(s): 5692ba5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -56,7 +56,7 @@ print(HF_TOKEN)
56
 
57
 
58
  m = AutoModelForCausalLM.from_pretrained(
59
- "ibm-granite/granite-3.0-2b-instruct",
60
  torch_dtype=torch.bfloat16 if torch.cuda.is_available() else torch.float32,
61
  trust_remote_code=True,token=HF_TOKEN
62
  )
 
56
 
57
 
58
  m = AutoModelForCausalLM.from_pretrained(
59
+ "google/gemma-2b-it",
60
  torch_dtype=torch.bfloat16 if torch.cuda.is_available() else torch.float32,
61
  trust_remote_code=True,token=HF_TOKEN
62
  )