safe model tensor updated

app.py

@@ -88,6 +88,8 @@ UNet_Encoder = UNet2DConditionModel_ref.from_pretrained(
     base_path,
     subfolder="unet_encoder",
     torch_dtype=torch.float16,
+    load_in_8bit=True,
+    max_memory=f'{int(torch.cuda.mem_get_info()[0]/1024**3)-2}GB'
 )
 
 parsing_model = Parsing(0)
@@ -120,8 +122,9 @@ pipe = TryonPipeline.from_pretrained(
         torch_dtype=torch.float16,
 )
 pipe.unet_encoder = UNet_Encoder
+pipe.to("cuda")
 
-@spaces.GPU
+@spaces.GPU(duration=120)
 def start_tryon(dict,garm_img,garment_des,is_checked,is_checked_crop,denoise_steps,seed):
     device = "cuda"
     

requirements.txt

@@ -21,7 +21,3 @@ fvcore
 cloudpickle
 omegaconf
 pycocotools
-gradio
-peft
-spaces
-bitsandbytes