rodrigomasini commited on
Commit
013f264
1 Parent(s): 5a405ec

Update app_v1.py

Browse files
Files changed (1) hide show
  1. app_v1.py +14 -1
app_v1.py CHANGED
@@ -8,7 +8,20 @@ import threading
8
  cwd = os.getcwd()
9
  cachedir = cwd+'/cache'
10
 
11
- # Assuming the rest of your setup code is correct and `local_folder` is properly set up
 
 
 
 
 
 
 
 
 
 
 
 
 
12
 
13
  class QuantizedModel:
14
  def __init__(self, model_dir):
 
8
  cwd = os.getcwd()
9
  cachedir = cwd+'/cache'
10
 
11
+ # Check if the directory exists before creating it
12
+ if not os.path.exists(cachedir):
13
+ os.mkdir(cachedir)
14
+
15
+ os.environ['HF_HOME'] = cachedir
16
+
17
+ local_folder = cachedir + "/model"
18
+
19
+
20
+ quantized_model_dir = "FPHam/Jackson_The_Formalizer_V2_13b_GPTQ"
21
+
22
+ snapshot_download(repo_id=quantized_model_dir, local_dir=local_folder, local_dir_use_symlinks=True)
23
+
24
+ model_basename = cachedir + "/model/Jackson2-4bit-128g-GPTQ"
25
 
26
  class QuantizedModel:
27
  def __init__(self, model_dir):