Futuretop commited on
Commit
32406a8
·
verified ·
1 Parent(s): 898a464

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ REVISION = "ceaf371f01ef66192264811b390bccad475a4f02"
9
 
10
  # Florence-2 로드
11
  device = "cuda" if torch.cuda.is_available() else "cpu"
12
- florence_model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base', revision = REVISION, trust_remote_code=True).to(device).eval()
13
  florence_processor = AutoProcessor.from_pretrained('microsoft/Florence-2-base', revision = REVISION, trust_remote_code=True)
14
 
15
  # Stable Diffusion TurboX 로드
 
9
 
10
  # Florence-2 로드
11
  device = "cuda" if torch.cuda.is_available() else "cpu"
12
+ florence_model = AutoModelForCausalLM.from_pretrained('microsoft/Florence-2-base', revision = REVISION, trust_remote_code=True, skip_init = True).to(device).eval()
13
  florence_processor = AutoProcessor.from_pretrained('microsoft/Florence-2-base', revision = REVISION, trust_remote_code=True)
14
 
15
  # Stable Diffusion TurboX 로드