codewithdark commited on
Commit
bee08bc
·
verified ·
1 Parent(s): 9c78cb3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ hf_client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
10
  local_model_name = "codewithdark/latent-recurrent-depth-lm"
11
  tokenizer = AutoTokenizer.from_pretrained(local_model_name)
12
  config = AutoConfig.from_pretrained(local_model_name)
13
- model = AutoModel.from_pretrained(local_model_name, config=config)
14
  device = "cuda" if torch.cuda.is_available() else "cpu"
15
  model.to(device)
16
 
 
10
  local_model_name = "codewithdark/latent-recurrent-depth-lm"
11
  tokenizer = AutoTokenizer.from_pretrained(local_model_name)
12
  config = AutoConfig.from_pretrained(local_model_name)
13
+ model = AutoModel.from_pretrained(local_model_name, config=config, trust_remote_code=True)
14
  device = "cuda" if torch.cuda.is_available() else "cpu"
15
  model.to(device)
16