gmkdigitalmedia commited on
Commit
c52ba39
·
1 Parent(s): a4a97b6

Fix PyTorch Docker uid error - set USER env var before imports

Browse files
Files changed (3) hide show
  1. app.py +7 -0
  2. foundation_engine.py +7 -1
  3. two_llm_system_FIXED.py +7 -0
app.py CHANGED
@@ -3,6 +3,13 @@ FastAPI REST API for Foundation 1.2 Clinical Trial System
3
  Production-ready Docker space with proper REST endpoints
4
  """
5
 
 
 
 
 
 
 
 
6
  from fastapi import FastAPI, HTTPException
7
  from fastapi.middleware.cors import CORSMiddleware
8
  from pydantic import BaseModel
 
3
  Production-ready Docker space with proper REST endpoints
4
  """
5
 
6
+ # Fix for PyTorch/Docker uid issue - must be before any torch imports
7
+ import os
8
+ os.environ["USER"] = "user"
9
+ os.environ["TORCH_HOME"] = "/tmp/torch"
10
+ os.environ["HF_HOME"] = "/tmp/huggingface"
11
+ os.environ["TRANSFORMERS_CACHE"] = "/tmp/transformers"
12
+
13
  from fastapi import FastAPI, HTTPException
14
  from fastapi.middleware.cors import CORSMiddleware
15
  from pydantic import BaseModel
foundation_engine.py CHANGED
@@ -3,8 +3,14 @@ Foundation 1.2
3
  Clinical trial query system with 355M foundation model
4
  """
5
 
6
- import gradio as gr
7
  import os
 
 
 
 
 
 
8
  from pathlib import Path
9
  import pickle
10
  import numpy as np
 
3
  Clinical trial query system with 355M foundation model
4
  """
5
 
6
+ # Fix for PyTorch/Docker uid issue - must be before any torch imports
7
  import os
8
+ os.environ["USER"] = "user"
9
+ os.environ["TORCH_HOME"] = "/tmp/torch"
10
+ os.environ["HF_HOME"] = "/tmp/huggingface"
11
+ os.environ["TRANSFORMERS_CACHE"] = "/tmp/transformers"
12
+
13
+ import gradio as gr
14
  from pathlib import Path
15
  import pickle
16
  import numpy as np
two_llm_system_FIXED.py CHANGED
@@ -4,6 +4,13 @@ Works with existing data structure: List[Tuple[int, str]]
4
  Keeps BM25 + semantic hybrid search intact
5
  """
6
 
 
 
 
 
 
 
 
7
  import torch
8
  from transformers import GPT2LMHeadModel, GPT2TokenizerFast, AutoTokenizer, AutoModelForCausalLM
9
  import logging
 
4
  Keeps BM25 + semantic hybrid search intact
5
  """
6
 
7
+ # Fix for PyTorch/Docker uid issue - must be before any torch imports
8
+ import os
9
+ os.environ.setdefault("USER", "user")
10
+ os.environ.setdefault("TORCH_HOME", "/tmp/torch")
11
+ os.environ.setdefault("HF_HOME", "/tmp/huggingface")
12
+ os.environ.setdefault("TRANSFORMERS_CACHE", "/tmp/transformers")
13
+
14
  import torch
15
  from transformers import GPT2LMHeadModel, GPT2TokenizerFast, AutoTokenizer, AutoModelForCausalLM
16
  import logging