Merge pull request #93 from intel/model_download_bug_fix

bug fix for sd-1.5 square model download

Author: gblong1

model_setup.py

@@ -74,15 +74,11 @@ def download_quantized_models(repo_id, model_fp16, model_int8):
             #print("download_folder", download_folder)
 
         FP16_model = os.path.join(download_folder, "FP16")
-        shutil.copytree(download_folder, SD_path_FP16, ignore=shutil.ignore_patterns('FP16', 'INT8'))    
+        shutil.copytree(download_folder, SD_path_FP16, ignore=shutil.ignore_patterns('FP16', 'INT8'))  
+        shutil.copytree(FP16_model, SD_path_FP16, dirs_exist_ok=True)        
+
 
 
-        files = glob(os.path.join(FP16_model, '**'), recursive=True)
-        
-        for f in files:
-            if os.path.isfile(f):
-                base = os.path.basename(f)
-                shutil.copy(f, os.path.join(SD_path_FP16, base))
 
         if model_int8:
             SD_path_INT8 = os.path.join(install_location, model_int8)           
@@ -91,15 +87,8 @@ def download_quantized_models(repo_id, model_fp16, model_int8):
                     shutil.rmtree(SD_path_INT8)
 
             INT8_model = os.path.join(download_folder, "INT8")
-            shutil.copytree(download_folder, SD_path_INT8, ignore=shutil.ignore_patterns('FP16', 'INT8'))        
-            #shutil.copy(INT8_model, SD_path_INT8)
-            
-            files = glob(os.path.join(INT8_model, '**'), recursive=True)
-            
-            for f in files:
-                if os.path.isfile(f):
-                    base = os.path.basename(f)
-                    shutil.copy(f, os.path.join(SD_path_INT8, base))          
+            shutil.copytree(download_folder, SD_path_INT8, ignore=shutil.ignore_patterns('FP16', 'INT8'))  
+            shutil.copytree(INT8_model, SD_path_INT8, dirs_exist_ok=True)
 
 
             delete_folder=os.path.join(download_folder, "..", "..", "..")