AlekseyCalvin commited on
Commit
16df3bd
·
verified ·
1 Parent(s): 003881c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -19,6 +19,12 @@ from huggingface_hub import login
19
 
20
  hf_token = os.environ.get("HF_TOKEN")
21
  login(token=hf_token)
 
 
 
 
 
 
22
 
23
  # Load LoRAs from JSON file
24
  with open('loras.json', 'r') as f:
 
19
 
20
  hf_token = os.environ.get("HF_TOKEN")
21
  login(token=hf_token)
22
+ cache_path = path.join(path.dirname(path.abspath(__file__)), "models")
23
+ os.environ["TRANSFORMERS_CACHE"] = cache_path
24
+ os.environ["HF_HUB_CACHE"] = cache_path
25
+ os.environ["HF_HOME"] = cache_path
26
+
27
+ torch.set_float32_matmul_precision("medium")
28
 
29
  # Load LoRAs from JSON file
30
  with open('loras.json', 'r') as f: