OmarAbualrob commited on
Commit
781fff6
·
verified ·
1 Parent(s): 7eb1af8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -37,14 +37,20 @@ async def startup_event():
37
  bnb_4bit_compute_dtype=torch_dtype
38
  )
39
 
 
40
  model = AutoModelForCausalLM.from_pretrained(
41
  model_id,
42
  trust_remote_code=True,
43
  quantization_config=quantization_config,
44
- revision="e134b72",
45
  )
46
 
47
- processor = AutoProcessor.from_pretrained(model_id, trust_remote_code=True, revision="e134b72")
 
 
 
 
 
48
 
49
  logger.info("Model and processor loaded successfully.")
50
 
 
37
  bnb_4bit_compute_dtype=torch_dtype
38
  )
39
 
40
+ # Load the model WITHOUT the invalid revision ID
41
  model = AutoModelForCausalLM.from_pretrained(
42
  model_id,
43
  trust_remote_code=True,
44
  quantization_config=quantization_config,
45
+ # revision="e134b72" <-- REMOVED THIS LINE
46
  )
47
 
48
+ # Load the processor WITHOUT the invalid revision ID
49
+ processor = AutoProcessor.from_pretrained(
50
+ model_id,
51
+ trust_remote_code=True
52
+ # revision="e134b72" <-- REMOVED THIS LINE
53
+ )
54
 
55
  logger.info("Model and processor loaded successfully.")
56