Futuretop commited on
Commit
c0f4255
·
verified ·
1 Parent(s): 19a2293

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -55,6 +55,7 @@ florence_processor = AutoProcessor.from_pretrained('microsoft/Florence-2-base',
55
  model_repo = "tensorart/stable-diffusion-3.5-large-TurboX"
56
  pipe = DiffusionPipeline.from_pretrained(
57
  model_repo,
 
58
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
59
  )
60
  pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(model_repo, subfolder="scheduler", shift=5)
 
55
  model_repo = "tensorart/stable-diffusion-3.5-large-TurboX"
56
  pipe = DiffusionPipeline.from_pretrained(
57
  model_repo,
58
+ trust_remote_code=True,
59
  torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
60
  )
61
  pipe.scheduler = FlowMatchEulerDiscreteScheduler.from_pretrained(model_repo, subfolder="scheduler", shift=5)