AminFaraji commited on
Commit
0b4e2db
·
verified ·
1 Parent(s): f703038

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -48,6 +48,7 @@ model = AutoModelForCausalLM.from_pretrained(
48
  MODEL_NAME, trust_remote_code=True, device_map="auto"
49
  )
50
  model = model.eval()
 
51
 
52
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
53
  print(f"Model device: {model.device}")
 
48
  MODEL_NAME, trust_remote_code=True, device_map="auto"
49
  )
50
  model = model.eval()
51
+ print('model loadeddddddddddddddddddddddd')
52
 
53
  tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
54
  print(f"Model device: {model.device}")