captainkyd commited on
Commit
d0a36d2
β€’
1 Parent(s): 31b6033

Update app.py

Browse files

added the quantization_config back in

Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -31,6 +31,13 @@ model = AutoModelForCausalLM.from_pretrained(
31
  quantization_config=quantization_config,
32
  )
33
 
 
 
 
 
 
 
 
34
  model_path = "WhiteRabbitNeo/Trinity-13B"
35
 
36
  hf_token = os.getenv("HF_TOKEN")
 
31
  quantization_config=quantization_config,
32
  )
33
 
34
+ model = AutoModelForCausalLM.from_pretrained(
35
+ model_path,
36
+ device_map="auto",
37
+ trust_remote_code=True,
38
+ quantization_config=quantization_config,
39
+ )
40
+
41
  model_path = "WhiteRabbitNeo/Trinity-13B"
42
 
43
  hf_token = os.getenv("HF_TOKEN")