Hugging Face's logo

Spaces:
shljessie
/
consistent_chat
Runtime error

App Files Community
1
shljessie commited on
Commit
9bb1dd2
·
1 Parent(s): f6f77b7

update model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ As a derivate work of [ConsistentAgents]() by Seonghee Lee.
16
  """
17
 
18
  if torch.cuda.is_available():
19
- model_id = "shljessie/profile-model-69"
20
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
21
  tokenizer = AutoTokenizer.from_pretrained(model_id)
22
  tokenizer.use_default_system_prompt = False
 
16
  """
17
 
18
  if torch.cuda.is_available():
19
+ model_id = "./backprop_llama2_69_1e-05"
20
  model = AutoModelForCausalLM.from_pretrained(model_id, torch_dtype=torch.float16, device_map="auto")
21
  tokenizer = AutoTokenizer.from_pretrained(model_id)
22
  tokenizer.use_default_system_prompt = False